Skip to content

Commit cf1db86

Browse files
author
Roja Reddy Sareddy
committed
update tensorflow artifacts
1 parent afbf206 commit cf1db86

File tree

9 files changed

+2
-47
lines changed

9 files changed

+2
-47
lines changed

tests/data/serve_resources/mlflow/tensorflow/MLmodel

Lines changed: 0 additions & 18 deletions
This file was deleted.

tests/data/serve_resources/mlflow/tensorflow/conda.yaml

Lines changed: 0 additions & 11 deletions
This file was deleted.

tests/data/serve_resources/mlflow/tensorflow/data/keras_module.txt

Lines changed: 0 additions & 1 deletion
This file was deleted.
-21.4 KB
Binary file not shown.

tests/data/serve_resources/mlflow/tensorflow/data/save_format.txt

Lines changed: 0 additions & 1 deletion
This file was deleted.

tests/data/serve_resources/mlflow/tensorflow/python_env.yaml

Lines changed: 0 additions & 7 deletions
This file was deleted.

tests/data/serve_resources/mlflow/tensorflow/requirements.txt

Lines changed: 0 additions & 4 deletions
This file was deleted.

tests/integ/sagemaker/serve/constants.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
SERVE_IN_PROCESS_TIMEOUT = 5
2121
SERVE_MODEL_PACKAGE_TIMEOUT = 10
2222
SERVE_LOCAL_CONTAINER_TIMEOUT = 10
23-
SERVE_SAGEMAKER_ENDPOINT_TIMEOUT = 15
23+
SERVE_SAGEMAKER_ENDPOINT_TIMEOUT = 50
2424
SERVE_SAVE_TIMEOUT = 2
2525

2626
PYTHON_VERSION_IS_NOT_38 = platform.python_version_tuple()[1] != "8"

tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -107,10 +107,7 @@ 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-
)
110+
114111
def test_happy_tensorflow_sagemaker_endpoint_with_tensorflow_serving(
115112
sagemaker_session,
116113
tensorflow_schema_builder,

0 commit comments

Comments
 (0)