We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4392a0c commit 588799aCopy full SHA for 588799a
tests/integ/test_horovod.py
@@ -37,6 +37,8 @@ def test_hvd_cpu(
37
cpu_instance_type,
38
tmpdir,
39
):
40
+ if Version(tensorflow_training_latest_version) >= Version("2.13"):
41
+ pytest.skip("Horovod is deprecated in TensorFlow 2.13 and above")
42
_create_and_fit_estimator(
43
sagemaker_session,
44
tensorflow_training_latest_version,
@@ -65,6 +67,8 @@ def test_hvd_gpu(
65
67
and kwargs["instance_type"] == "ml.p2.xlarge"
66
68
69
pytest.skip("P2 instances have been deprecated for sagemaker jobs starting TensorFlow 2.12")
70
71
72
73
74
0 commit comments