Skip to content

Commit a19f57e

Browse files
author
Roja Reddy Sareddy
committed
Fix incompatible_dependecies test
1 parent ebf56b9 commit a19f57e

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

tests/integ/sagemaker/remote_function/test_decorator.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,10 @@ def divide(x, y):
315315
divide(10, 2)
316316

317317

318+
@pytest.mark.skipif(
319+
np.__version__ >= "2.0",
320+
reason="Test only valid for numpy < 2.0 due to serialization compatibility changes",
321+
)
318322
def test_with_incompatible_dependencies(
319323
sagemaker_session, dummy_container_without_error, cpu_instance_type
320324
):
@@ -323,7 +327,8 @@ def test_with_incompatible_dependencies(
323327
It is currently working with python 3.8+. However, running it with older versions
324328
or versions in the future may require changes to 'old_deps_requirements.txt'
325329
to fulfill testing scenario.
326-
330+
331+
NOTE: Skipped for numpy >= 2.0 as serialization compatibility improved.
327332
"""
328333

329334
dependencies_path = os.path.join(DATA_DIR, "remote_function", "old_deps_requirements.txt")

tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,10 @@ def tensorflow_schema_builder(custom_request_translator, custom_response_transla
107107
PYTHON_VERSION_IS_NOT_310,
108108
reason="The goal of these test are to test the serving components of our feature",
109109
)
110+
@pytest.mark.skipif(
111+
np.__version__ >= "2.0",
112+
reason="TensorFlow serving container not yet compatible with numpy 2.0 serialization",
113+
)
110114
def test_happy_tensorflow_sagemaker_endpoint_with_tensorflow_serving(
111115
sagemaker_session,
112116
tensorflow_schema_builder,

0 commit comments

Comments
 (0)