Skip to content

Commit 588799a

Browse files
author
Mike Schneider
committed
skip hvd tests and tf 2.13 does not have hvd.
1 parent 4392a0c commit 588799a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/integ/test_horovod.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ def test_hvd_cpu(
3737
cpu_instance_type,
3838
tmpdir,
3939
):
40+
if Version(tensorflow_training_latest_version) >= Version("2.13"):
41+
pytest.skip("Horovod is deprecated in TensorFlow 2.13 and above")
4042
_create_and_fit_estimator(
4143
sagemaker_session,
4244
tensorflow_training_latest_version,
@@ -65,6 +67,8 @@ def test_hvd_gpu(
6567
and kwargs["instance_type"] == "ml.p2.xlarge"
6668
):
6769
pytest.skip("P2 instances have been deprecated for sagemaker jobs starting TensorFlow 2.12")
70+
if Version(tensorflow_training_latest_version) >= Version("2.13"):
71+
pytest.skip("Horovod is deprecated in TensorFlow 2.13 and above")
6872

6973
_create_and_fit_estimator(
7074
sagemaker_session,

0 commit comments

Comments
 (0)