Skip to content

Commit c6ebb5d

Browse files
committed
Fixing unit tests for MWMS
1 parent 2c48152 commit c6ebb5d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/sagemaker/tensorflow/test_estimator.py

+2
Original file line numberDiff line numberDiff line change
@@ -526,6 +526,8 @@ def test_fit_mpi(time, strftime, sagemaker_session):
526526
def test_fit_mwms(
527527
time, strftime, sagemaker_session, tensorflow_training_version, tensorflow_training_py_version
528528
):
529+
if version.Version(tensorflow_training_version) < version.Version("2.11"):
530+
pytest.skip("Multi Worker Mirrored Strategy was added in TF 2.11")
529531
framework_version = tensorflow_training_version
530532
py_version = tensorflow_training_py_version
531533
tf = TensorFlow(

0 commit comments

Comments
 (0)