Skip to content

Commit 46d069d

Browse files
committed
fix: skip TF tests for unsupported versions
1 parent 92b706a commit 46d069d

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

tests/integ/sagemaker/workflow/test_model_create_and_registration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1422,7 +1422,7 @@ def test_model_registration_with_tensorflow_model_with_pipeline_model(
14221422
pipeline_name,
14231423
region_name,
14241424
):
1425-
if Version(tf_full_version) in SpecifierSet("==2.16.*"):
1425+
if Version(tf_full_version) >= Version("2.16"):
14261426
pytest.skip(
14271427
"This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
14281428
"https://github.com/tensorflow/io/issues/2039"

tests/integ/sagemaker/workflow/test_model_steps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -592,7 +592,7 @@ def test_model_registration_with_drift_check_baselines_and_model_metrics(
592592
def test_model_registration_with_tensorflow_model_with_pipeline_model(
593593
pipeline_session, role, tf_full_version, tf_full_py_version, pipeline_name
594594
):
595-
if Version(tf_full_version) in SpecifierSet("==2.16.*"):
595+
if Version(tf_full_version) >= Version("2.16"):
596596
pytest.skip(
597597
"This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
598598
"https://github.com/tensorflow/io/issues/2039"

tests/integ/sagemaker/workflow/test_training_steps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ def test_training_step_with_output_path_as_join(
238238
def test_tensorflow_training_step_with_parameterized_code_input(
239239
pipeline_session, role, tf_full_version, tf_full_py_version, pipeline_name
240240
):
241-
if Version(tf_full_version) in SpecifierSet("==2.16.*"):
241+
if Version(tf_full_version) >= Version("2.16"):
242242
pytest.skip(
243243
"This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
244244
"https://github.com/tensorflow/io/issues/2039"

tests/integ/test_transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -556,7 +556,7 @@ def test_transform_mxnet_logs(
556556
def test_transform_tf_kms_network_isolation(
557557
sagemaker_session, cpu_instance_type, tmpdir, tf_full_version, tf_full_py_version
558558
):
559-
if Version(tf_full_version) in SpecifierSet("==2.16.*"):
559+
if Version(tf_full_version) >= Version("2.16"):
560560
pytest.skip(
561561
"This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
562562
"https://github.com/tensorflow/io/issues/2039"

0 commit comments

Comments
 (0)