Skip to content

Commit 9a7ced6

Browse files
committed
Correct TF 2.7 patch version used in tests
1 parent af57e4e commit 9a7ced6

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/unit/test_fw_utils.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -686,7 +686,7 @@ def test_validate_smdataparallel_args_not_raises():
686686
("ml.p3.16xlarge", "tensorflow", "2.6.0", "py38", smdataparallel_enabled),
687687
("ml.p3.16xlarge", "tensorflow", "2.6.2", "py38", smdataparallel_enabled),
688688
("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),
690690
("ml.p3.16xlarge", "tensorflow", "2.7", "py38", smdataparallel_enabled),
691691
("ml.p3.16xlarge", "tensorflow", "2.8.0", "py38", smdataparallel_enabled),
692692
("ml.p3.16xlarge", "tensorflow", "2.8", "py39", smdataparallel_enabled),
@@ -700,11 +700,11 @@ def test_validate_smdataparallel_args_not_raises():
700700
("ml.p3.16xlarge", "pytorch", "1.9.1", "py38", smdataparallel_enabled),
701701
("ml.p3.16xlarge", "pytorch", "1.9", "py38", smdataparallel_enabled),
702702
("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),
705705
("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),
708708
("ml.p3.16xlarge", "tensorflow", "2.8.0", "py38", smdataparallel_enabled_custom_mpi),
709709
("ml.p3.16xlarge", "pytorch", "1.8.0", "py3", smdataparallel_enabled_custom_mpi),
710710
("ml.p3.16xlarge", "pytorch", "1.9.1", "py38", smdataparallel_enabled_custom_mpi),

0 commit comments

Comments
 (0)