numpy fixes

Roja Reddy Sareddy · Roja Reddy Sareddy · commit fc64ae62c4e9 · 2025-10-08T05:33:51.000-07:00
diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt
@@ -44,7 +44,7 @@ onnx==1.17.0
 nbformat>=5.9,<6
 accelerate>=0.24.1,<=0.27.0
 schema==0.7.5
-tensorflow>=2.16.2,<=2.18.0
+tensorflow>=2.16.2,<=2.19.0
 mlflow>=2.14.2,<3
 huggingface_hub==0.26.2
 uvicorn>=0.30.1
diff --git a/tests/data/serve_resources/mlflow/tensorflow/MLmodel b/tests/data/serve_resources/mlflow/tensorflow/MLmodel
@@ -12,6 +12,5 @@ flavors:
     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'
+model_uuid: 129070f64c7b4ba4b24971c1840947e7
+utc_time_created: '2025-09-30 03:13:23.791740'
diff --git a/tests/data/serve_resources/mlflow/tensorflow/conda.yaml b/tests/data/serve_resources/mlflow/tensorflow/conda.yaml
@@ -4,8 +4,8 @@ dependencies:
 - python=3.10.13
 - pip<=24.3
 - pip:
-  - mlflow>=2.16.1
+  - mlflow==2.20.3
   - cloudpickle>=2.2.1
-  - numpy>=1.26.4
-  - tensorflow==2.18.0
-name: mlflow-env
+  - numpy>=2.0.0
+  - tensorflow==2.19.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
@@ -4,4 +4,4 @@ build_dependencies:
 - setuptools==68.2.2
 - wheel==0.41.2
 dependencies:
-- -r requirements.txt
+- -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
@@ -1,2 +1 @@
-model_name: model
-model_version: '2'
+registered_model_name: numpy2-test-model
diff --git a/tests/data/serve_resources/mlflow/tensorflow/requirements.txt b/tests/data/serve_resources/mlflow/tensorflow/requirements.txt
@@ -1,4 +1,7 @@
 mlflow==2.20.3
 cloudpickle>=2.2.1
-numpy>=1.26.4
-tensorflow==2.18.0
+numpy>=2.0.0,<3.0.0
+tensorflow==2.19.0
+scipy==1.13.0
+scikit-learn==1.6.1
+pandas==2.3.0
diff --git a/tests/data/serve_resources/mlflow/tensorflow/tf2model/fingerprint.pb b/tests/data/serve_resources/mlflow/tensorflow/tf2model/fingerprint.pb
@@ -1 +1 @@
-���������ď����n�����/ �ʢ������(��������32
+֕��������ۻ�ҍ�����������m ��Ę���:(�ࢿ�ǔ�J2:4997811417122571964
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
@@ -105,7 +105,7 @@ def tensorflow_schema_builder(custom_request_translator, custom_response_transla
 
 @pytest.mark.skipif(
     PYTHON_VERSION_IS_NOT_310,
-    np.__version__ >= "2.0.0",
+    #np.__version__ >= "2.0.0",
     reason="The goal of these test are to test the serving components of our feature and \
            the input model artifacts used here are generated with py310 and numpy<2",
 )

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1 @@`
`1`		`-model_name: model`
`2`		`-model_version: '2'`
	`1`	`+registered_model_name: numpy2-test-model`
Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-��ď��n����/ �ʢ��(��32`
	`1`	`+֕��ۻ�ҍ��m ��Ę��:(�ࢿ�ǔ�J2:4997811417122571964`