Add numpy 2.0 support

Author: Roja Reddy Sareddy

requirements/extras/test_requirements.txt

@@ -1,5 +1,5 @@
 tox==3.24.5
-numpy>=1.26.4, <2.0
+numpy>=2.0.0, <3.0
 build[virtualenv]==1.2.1
 flake8==7.1.2
 pytest==6.2.5

src/sagemaker/serve/utils/conda_in_process.yml

@@ -12,7 +12,7 @@ dependencies:
     - boto3>=1.34.142,<2.0
     - cloudpickle==2.2.1
     - google-pasta
-    - numpy>=1.26.4,<2.0
+    - numpy>=2.0.0,<3.0
     - protobuf>=3.12,<5.0
     - smdebug_rulesconfig==1.0.1
     - importlib-metadata>=1.4.0,<7.0

tests/data/remote_function/requirements.txt

@@ -1 +1 @@
-scipy==1.11.3
+scipy==1.13.0

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

@@ -11,7 +11,7 @@ dependencies:
   - defusedxml==0.7.1
   - dill==0.3.9
   - gmpy2==2.1.2
-  - numpy>=1.26.4,<2.0
+  - numpy>=2.0.0,<3.0
   - opt-einsum==3.3.0
   - packaging==24.0
   - pandas>=2.3.0

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

@@ -5,7 +5,7 @@ cloudpickle==2.2.1
 defusedxml==0.7.1
 dill==0.3.9
 gmpy2==2.1.2
-numpy>=1.26.4,<2.0
+numpy>=2.0.0,<3.0
 opt-einsum==3.3.0
 packaging>=23.0,<25
 pandas>=2.3.0

tests/data/serve_resources/mlflow/tensorflow/MLmodel

@@ -10,7 +10,7 @@ flavors:
     code: null
     model_type: tf2-module
     saved_model_dir: tf2model
-mlflow_version: 2.20.3
+mlflow_version: 2.11.1
 model_size_bytes: 23823
 model_uuid: 40d2323944294fce898d8693455f60e8
 run_id: 592132312fb84935b201de2c027c54c6

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

@@ -1,8 +1,8 @@
 mlflow==3.1.0
 lz4==4.3.2
-numpy>=1.26.4,<2.0
+numpy>=1.26.4,<3.0
 pandas>=2.3.0
 psutil==5.9.8
-scikit-learn==1.5.1
-scipy==1.11.3
+scikit-learn==1.6.1
+scipy==1.13.0
 xgboost==1.7.1

tests/integ/sagemaker/experiments/test_run.py

@@ -693,7 +693,7 @@ def _generate_estimator(
         sagemaker_client_config=sagemaker_client_config,
     )
     return SKLearn(
-        framework_version="1.2-1",
+        framework_version="1.4-2",
         entry_point=_ENTRY_POINT_PATH,
         dependencies=[sdk_tar],
         role=execution_role,

tests/unit/sagemaker/jumpstart/constants.py

@@ -12095,7 +12095,7 @@
         "inference_vulnerabilities": [],
         "training_vulnerable": False,
         "training_dependencies": [
-            "numpy>=1.26.4",
+            "numpy>=2.0.0",
             "opencv_python==4.7.0.68",
             "sagemaker_jumpstart_prepack_script_utilities==1.0.0",
         ],
@@ -14360,7 +14360,7 @@
             "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.3.0",