@@ -686,7 +686,7 @@ def test_validate_smdataparallel_args_not_raises():
686
686
("ml.p3.16xlarge" , "tensorflow" , "2.6.0" , "py38" , smdataparallel_enabled ),
687
687
("ml.p3.16xlarge" , "tensorflow" , "2.6.2" , "py38" , smdataparallel_enabled ),
688
688
("ml.p3.16xlarge" , "tensorflow" , "2.6" , "py38" , smdataparallel_enabled ),
689
- ("ml.p3.16xlarge" , "tensorflow" , "2.7.0 " , "py38" , smdataparallel_enabled ),
689
+ ("ml.p3.16xlarge" , "tensorflow" , "2.7.1 " , "py38" , smdataparallel_enabled ),
690
690
("ml.p3.16xlarge" , "tensorflow" , "2.7" , "py38" , smdataparallel_enabled ),
691
691
("ml.p3.16xlarge" , "tensorflow" , "2.8.0" , "py38" , smdataparallel_enabled ),
692
692
("ml.p3.16xlarge" , "tensorflow" , "2.8" , "py39" , smdataparallel_enabled ),
@@ -700,11 +700,11 @@ def test_validate_smdataparallel_args_not_raises():
700
700
("ml.p3.16xlarge" , "pytorch" , "1.9.1" , "py38" , smdataparallel_enabled ),
701
701
("ml.p3.16xlarge" , "pytorch" , "1.9" , "py38" , smdataparallel_enabled ),
702
702
("ml.p3.16xlarge" , "pytorch" , "1.10" , "py38" , smdataparallel_enabled ),
703
- ("ml.p3.16xlarge" , "tensorflow" , "2.4.1 " , "py3" , smdataparallel_enabled_custom_mpi ),
704
- ("ml.p3.16xlarge" , "tensorflow" , "2.4.1 " , "py37" , smdataparallel_enabled_custom_mpi ),
703
+ ("ml.p3.16xlarge" , "tensorflow" , "2.4.3 " , "py3" , smdataparallel_enabled_custom_mpi ),
704
+ ("ml.p3.16xlarge" , "tensorflow" , "2.4.3 " , "py37" , smdataparallel_enabled_custom_mpi ),
705
705
("ml.p3.16xlarge" , "tensorflow" , "2.5.1" , "py37" , smdataparallel_enabled_custom_mpi ),
706
- ("ml.p3.16xlarge" , "tensorflow" , "2.6.0 " , "py38" , smdataparallel_enabled_custom_mpi ),
707
- ("ml.p3.16xlarge" , "tensorflow" , "2.7.0 " , "py38" , smdataparallel_enabled_custom_mpi ),
706
+ ("ml.p3.16xlarge" , "tensorflow" , "2.6.2 " , "py38" , smdataparallel_enabled_custom_mpi ),
707
+ ("ml.p3.16xlarge" , "tensorflow" , "2.7.1 " , "py38" , smdataparallel_enabled_custom_mpi ),
708
708
("ml.p3.16xlarge" , "tensorflow" , "2.8.0" , "py38" , smdataparallel_enabled_custom_mpi ),
709
709
("ml.p3.16xlarge" , "pytorch" , "1.8.0" , "py3" , smdataparallel_enabled_custom_mpi ),
710
710
("ml.p3.16xlarge" , "pytorch" , "1.9.1" , "py38" , smdataparallel_enabled_custom_mpi ),
0 commit comments