resolve conflict dependency with numpy 2.0

Roja Reddy Sareddy · Roja Reddy Sareddy · commit 5817b2e51501 · 2025-04-17T23:33:53.000-07:00
diff --git a/tests/data/remote_function/old_deps_requirements.txt b/tests/data/remote_function/old_deps_requirements.txt
@@ -1 +1 @@
-pandas==1.3.4
+pandas==2.2.3
diff --git a/tests/data/remote_function/requirements.txt b/tests/data/remote_function/requirements.txt
@@ -1 +1 @@
-scipy==1.11.3
+scipy==1.13.0
diff --git a/tests/data/serve_resources/mlflow/pytorch/conda.yaml b/tests/data/serve_resources/mlflow/pytorch/conda.yaml
@@ -14,7 +14,7 @@ dependencies:
   - numpy>=2.0.0,<2.3.0
   - opt-einsum==3.3.0
   - packaging==24.0
-  - pandas==2.2.1
+  - pandas==2.2.3
   - pyyaml==6.0.1
   - requests==2.31.0
   - torch>=2.6.0
diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
@@ -8,7 +8,7 @@ gmpy2==2.1.2
 numpy>=2.0.0,<2.3.0
 opt-einsum==3.3.0
 packaging>=23.0,<25
-pandas==2.2.1
+pandas==2.2.3
 pyyaml==6.0.1
 requests==2.32.2
 torch>=2.6.0
diff --git a/tests/data/serve_resources/mlflow/tensorflow/conda.yaml b/tests/data/serve_resources/mlflow/tensorflow/conda.yaml
@@ -7,5 +7,5 @@ dependencies:
   - mlflow==2.16.1
   - cloudpickle==2.2.1
   - numpy>=2.0.0,<2.3.0
-  - tensorflow==2.16.1
+  - tensorflow==2.17.0
 name: mlflow-env
diff --git a/tests/data/serve_resources/mlflow/tensorflow/requirements.txt b/tests/data/serve_resources/mlflow/tensorflow/requirements.txt
@@ -1,4 +1,4 @@
 mlflow==2.16.1
 cloudpickle==2.2.1
 numpy>=2.0.0,<2.3.0
-tensorflow==2.16.1
+tensorflow==2.17.0
diff --git a/tests/data/serve_resources/mlflow/xgboost/conda.yaml b/tests/data/serve_resources/mlflow/xgboost/conda.yaml
@@ -7,9 +7,9 @@ dependencies:
   - mlflow==2.16.1
   - lz4==4.3.2
   - numpy>=2.0.0,<2.3.0
-  - pandas==2.2.1
+  - pandas==2.2.3
   - psutil==5.9.8
   - scikit-learn==1.6.1
-  - scipy==1.11.3
+  - scipy==1.13.0
   - xgboost==1.7.1
 name: mlflow-env
diff --git a/tests/data/serve_resources/mlflow/xgboost/requirements.txt b/tests/data/serve_resources/mlflow/xgboost/requirements.txt
@@ -1,8 +1,8 @@
 mlflow==2.16.1
 lz4==4.3.2
 numpy>=2.0.0,<2.3.0
-pandas==2.0.3
+pandas==2.2.3
 psutil==5.9.8
 scikit-learn==1.6.1
-scipy==1.11.3
+scipy==1.13.0
 xgboost==1.7.1
diff --git a/tests/data/workflow/requirements.txt b/tests/data/workflow/requirements.txt
@@ -1 +1 @@
-scipy==1.11.3
+scipy==1.13.0
diff --git a/tests/unit/sagemaker/jumpstart/constants.py b/tests/unit/sagemaker/jumpstart/constants.py
@@ -5361,7 +5361,7 @@
             "safetensors==0.3.1",
             "sagemaker_jumpstart_huggingface_script_utilities==1.1.3",
             "sagemaker_jumpstart_script_utilities==1.1.9",
-            "scipy==1.11.1",
+            "scipy==1.13.0",
             "termcolor==2.3.0",
             "texttable==1.6.7",
             "tokenize-rt==5.1.0",
@@ -7870,7 +7870,7 @@
             "safetensors==0.3.1",
             "sagemaker_jumpstart_huggingface_script_utilities==1.1.3",
             "sagemaker_jumpstart_script_utilities==1.1.9",
-            "scipy==1.11.1",
+            "scipy==1.13.0",
             "termcolor==2.3.0",
             "texttable==1.6.7",
             "tokenize-rt==5.1.0",
@@ -8346,7 +8346,7 @@
             "safetensors==0.3.1",
             "sagemaker_jumpstart_huggingface_script_utilities==1.1.3",
             "sagemaker_jumpstart_script_utilities==1.1.9",
-            "scipy==1.11.1",
+            "scipy==1.13.0",
             "termcolor==2.3.0",
             "texttable==1.6.7",
             "tokenize-rt==5.1.0",
@@ -12095,7 +12095,7 @@
         "inference_vulnerabilities": [],
         "training_vulnerable": False,
         "training_dependencies": [
-            "numpy==1.23.1",
+            "numpy>=2.0.0",
             "opencv_python==4.7.0.68",
             "sagemaker_jumpstart_prepack_script_utilities==1.0.0",
         ],
@@ -14360,10 +14360,10 @@
             "jmespath==1.0.1",
             "jsonschema==4.17.3",
             "multiprocess==0.70.14",
-            "numpy==1.26.4",
+            "numpy>=2.0.0",
             "oscrypto==1.3.0",
             "packaging==23.1",
-            "pandas==2.0.2",
+            "pandas==2.2.3",
             "pathos==0.3.0",
             "pkgutil-resolve-name==1.3.10",
             "platformdirs==3.8.0",
@@ -14884,10 +14884,10 @@
             "jmespath==1.0.1",
             "jsonschema==4.17.3",
             "multiprocess==0.70.14",
-            "numpy==1.24.3",
+            "numpy>=2.0.0",
             "oscrypto==1.3.0",
             "packaging==23.1",
-            "pandas==2.0.2",
+            "pandas==2.2.3",
             "pathos==0.3.0",
             "pkgutil-resolve-name==1.3.10",
             "platformdirs==3.8.0",
@@ -17400,7 +17400,7 @@
             "safetensors==0.3.1",
             "sagemaker_jumpstart_huggingface_script_utilities==1.1.4",
             "sagemaker_jumpstart_script_utilities==1.1.9",
-            "scipy==1.11.1",
+            "scipy==1.13.0",
             "termcolor==2.3.0",
             "texttable==1.6.7",
             "tokenize-rt==5.1.0",
diff --git a/tests/unit/sagemaker/serve/detector/test_dependency_manager.py b/tests/unit/sagemaker/serve/detector/test_dependency_manager.py
@@ -21,7 +21,7 @@
 
 DEPENDENCY_LIST = [
     "requests==2.26.0",
-    "numpy==1.26.4",
+    "numpy>=2.0.0",
     "pandas<=1.3.3",
     "matplotlib<3.5.0",
     "scikit-learn>0.24.1",
