adding PT 1.13.1 to test_fw_utils

Mike Schneider · Mike Schneider · commit d598c07b9312 · 2023-01-24T21:07:34.000Z
diff --git a/tests/unit/test_fw_utils.py b/tests/unit/test_fw_utils.py
@@ -932,6 +932,7 @@ def test_validate_smdataparallel_args_not_raises():
         ("ml.p3.16xlarge", "pytorch", "1.11.0", "py38", smdataparallel_enabled_custom_mpi),
         ("ml.p3.16xlarge", "pytorch", "1.12.0", "py38", smdataparallel_enabled_custom_mpi),
         ("ml.p3.16xlarge", "pytorch", "1.12.1", "py38", smdataparallel_enabled_custom_mpi),
+        ("ml.p3.16xlarge", "pytorch", "1.13.1", "py39", smdataparallel_enabled_custom_mpi),
     ]
     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
`@@ -932,6 +932,7 @@ def test_validate_smdataparallel_args_not_raises():`
`932`	`932`	`("ml.p3.16xlarge", "pytorch", "1.11.0", "py38", smdataparallel_enabled_custom_mpi),`
`933`	`933`	`("ml.p3.16xlarge", "pytorch", "1.12.0", "py38", smdataparallel_enabled_custom_mpi),`
`934`	`934`	`("ml.p3.16xlarge", "pytorch", "1.12.1", "py38", smdataparallel_enabled_custom_mpi),`
	`935`	`+ ("ml.p3.16xlarge", "pytorch", "1.13.1", "py39", smdataparallel_enabled_custom_mpi),`
`935`	`936`	`]`
`936`	`937`	`for instance_type, framework_name, framework_version, py_version, distribution in good_args:`
`937`	`938`	`fw_utils._validate_smdataparallel_args(`