add test for pt171 check

ChaiBapchya · ChaiBapchya · commit bcbca31f6aee · 2020-12-14T13:46:01.000-08:00
diff --git a/tests/unit/test_fw_utils.py b/tests/unit/test_fw_utils.py
@@ -632,6 +632,7 @@ def test_validate_smdataparallel_args_not_raises():
         (None, None, None, None, smdataparallel_disabled),
         ("ml.p3.16xlarge", "tensorflow", "2.3.1", "py3", smdataparallel_enabled),
         ("ml.p3.16xlarge", "pytorch", "1.6.0", "py3", smdataparallel_enabled),
+        ("ml.p3.16xlarge", "pytorch", "1.7.0", "py3", smdataparallel_enabled),
     ]
     for instance_type, framework_name, framework_version, py_version, distribution in good_args:
         fw_utils._validate_smdataparallel_args(

Original file line number	Diff line number	Diff line change
`@@ -632,6 +632,7 @@ def test_validate_smdataparallel_args_not_raises():`
`632`	`632`	`(None, None, None, None, smdataparallel_disabled),`
`633`	`633`	`("ml.p3.16xlarge", "tensorflow", "2.3.1", "py3", smdataparallel_enabled),`
`634`	`634`	`("ml.p3.16xlarge", "pytorch", "1.6.0", "py3", smdataparallel_enabled),`
	`635`	`+ ("ml.p3.16xlarge", "pytorch", "1.7.0", "py3", smdataparallel_enabled),`
`635`	`636`	`]`
`636`	`637`	`for instance_type, framework_name, framework_version, py_version, distribution in good_args:`
`637`	`638`	`fw_utils._validate_smdataparallel_args(`