Skip tests with deprecated instance type (#5077)

Co-authored-by: pintaoz <[email protected]>

Author: pintaoz-aws

tests/integ/test_horovod.py

@@ -62,11 +62,8 @@ def test_hvd_gpu(
     tmpdir,
     **kwargs,
 ):
-    if (
-        Version(tensorflow_training_latest_version) >= Version("2.12")
-        and kwargs["instance_type"] == "ml.p2.xlarge"
-    ):
-        pytest.skip("P2 instances have been deprecated for sagemaker jobs starting TensorFlow 2.12")
+    if kwargs["instance_type"] == "ml.p2.xlarge":
+        pytest.skip("Instance type ml.p2.xlarge has been deprecated")
     if Version(tensorflow_training_latest_version) >= Version("2.13"):
         pytest.skip("Horovod is deprecated in TensorFlow 2.13 and above")
 

tests/integ/test_horovod_mx.py

@@ -58,6 +58,9 @@ def test_hvd_gpu(
     tmpdir,
     **kwargs,
 ):
+    if kwargs["instance_type"] == "ml.p2.xlarge":
+        pytest.skip("Instance type ml.p2.xlarge has been deprecated")
+
     _create_and_fit_estimator(
         mxnet_training_latest_version,
         mxnet_training_latest_py_version,