testfile codestyle fixes

Roja Reddy Sareddy · Roja Reddy Sareddy · commit 9884e7ba1886 · 2025-10-07T11:00:41.000-07:00
diff --git a/tests/data/serve_resources/mlflow/tensorflow/MLmodel b/tests/data/serve_resources/mlflow/tensorflow/MLmodel
@@ -0,0 +1,17 @@
+artifact_path: model
+flavors:
+  python_function:
+    env:
+      conda: conda.yaml
+      virtualenv: python_env.yaml
+    loader_module: mlflow.tensorflow
+    python_version: 3.10.13
+  tensorflow:
+    code: null
+    model_type: tf2-module
+    saved_model_dir: tf2model
+mlflow_version: 2.20.3
+model_size_bytes: 23823
+model_uuid: 40d2323944294fce898d8693455f60e8
+run_id: 592132312fb84935b201de2c027c54c6
+utc_time_created: '2024-04-01 19:47:15.396517'
diff --git a/tests/data/serve_resources/mlflow/tensorflow/conda.yaml b/tests/data/serve_resources/mlflow/tensorflow/conda.yaml
@@ -0,0 +1,11 @@
+channels:
+- conda-forge
+dependencies:
+- python=3.10.13
+- pip<=24.3
+- pip:
+  - mlflow>=2.16.1
+  - cloudpickle>=2.2.1
+  - numpy>=1.26.4
+  - tensorflow==2.18.0
+name: mlflow-env
diff --git a/tests/data/serve_resources/mlflow/tensorflow/python_env.yaml b/tests/data/serve_resources/mlflow/tensorflow/python_env.yaml
@@ -0,0 +1,7 @@
+python: 3.10.13
+build_dependencies:
+- pip==23.3.1
+- setuptools==68.2.2
+- wheel==0.41.2
+dependencies:
+- -r requirements.txt
diff --git a/tests/data/serve_resources/mlflow/tensorflow/registered_model_meta b/tests/data/serve_resources/mlflow/tensorflow/registered_model_meta
@@ -0,0 +1,2 @@
+model_name: model
+model_version: '2'
diff --git a/tests/data/serve_resources/mlflow/tensorflow/requirements.txt b/tests/data/serve_resources/mlflow/tensorflow/requirements.txt
@@ -0,0 +1,4 @@
+mlflow==2.20.3
+cloudpickle>=2.2.1
+numpy>=1.26.4
+tensorflow==2.18.0
diff --git a/tests/data/serve_resources/mlflow/tensorflow/tf2model/fingerprint.pb b/tests/data/serve_resources/mlflow/tensorflow/tf2model/fingerprint.pb
@@ -0,0 +1 @@
+���������ď����n�����/ �ʢ������(��������32
diff --git a/tests/data/serve_resources/mlflow/tensorflow/tf2model/saved_model.pb b/tests/data/serve_resources/mlflow/tensorflow/tf2model/saved_model.pb
diff --git a/tests/data/serve_resources/mlflow/tensorflow/tf2model/variables/variables.data-00000-of-00001 b/tests/data/serve_resources/mlflow/tensorflow/tf2model/variables/variables.data-00000-of-00001
diff --git a/tests/data/serve_resources/mlflow/tensorflow/tf2model/variables/variables.index b/tests/data/serve_resources/mlflow/tensorflow/tf2model/variables/variables.index
diff --git a/tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py b/tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py
@@ -37,7 +37,6 @@
 from sagemaker.serve.utils.lineage_constants import (
     MODEL_BUILDER_MLFLOW_MODEL_PATH_LINEAGE_ARTIFACT_TYPE,
 )
-import numpy as np
 
 logger = logging.getLogger(__name__)
 

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+model_name: model`
	`2`	`+model_version: '2'`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+��ď��n����/ �ʢ��(��32`
Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,6 @@`
`37`	`37`	`from sagemaker.serve.utils.lineage_constants import (`
`38`	`38`	`MODEL_BUILDER_MLFLOW_MODEL_PATH_LINEAGE_ARTIFACT_TYPE,`
`39`	`39`	`)`
`40`		`-import numpy as np`
`41`	`40`
`42`	`41`	`logger = logging.getLogger(__name__)`
`43`	`42`