Skip to content

Commit e4b80c1

Browse files
author
Roja Reddy Sareddy
committed
testfile codestyle fixes
1 parent 5b3766b commit e4b80c1

File tree

8 files changed

+2
-43
lines changed

8 files changed

+2
-43
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/test_serve_mlflow_tensorflow_flavor_happy.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
from sagemaker.serve.utils.lineage_constants import (
3838
MODEL_BUILDER_MLFLOW_MODEL_PATH_LINEAGE_ARTIFACT_TYPE,
3939
)
40+
import numpy as np
4041

4142
logger = logging.getLogger(__name__)
4243

@@ -105,9 +106,9 @@ def tensorflow_schema_builder(custom_request_translator, custom_response_transla
105106

106107
@pytest.mark.skipif(
107108
PYTHON_VERSION_IS_NOT_310,
109+
np.__version__ >= "2.0.0",
108110
reason="The goal of these test are to test the serving components of our feature",
109111
)
110-
111112
def test_happy_tensorflow_sagemaker_endpoint_with_tensorflow_serving(
112113
sagemaker_session,
113114
tensorflow_schema_builder,

0 commit comments

Comments
 (0)