Add numpy 2.0 support

Author: Roja Reddy Sareddy

pyproject.toml

@@ -39,7 +39,7 @@ dependencies = [
   "google-pasta",
   "importlib-metadata>=1.4.0,<7.0",
   "jsonschema",
-  "numpy==1.26.4",
+  "numpy>=2.0.0,<2.3.0",
   "omegaconf>=2.2,<3",
   "packaging>=23.0,<25",
   "pandas",

requirements/extras/scipy_requirements.txt

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

requirements/extras/test_requirements.txt

@@ -1,5 +1,5 @@
 tox==3.24.5
-numpy==1.26.4
+numpy>=2.0.0, <2.3.0
 build[virtualenv]==1.2.1
 flake8==7.1.2
 pytest==6.2.5
@@ -23,8 +23,8 @@ requests==2.32.2
 sagemaker-experiments==0.1.35
 Jinja2==3.1.6
 pyvis==0.2.1
-pandas==1.4.4
-scikit-learn==1.3.0
+pandas==2.2.3
+scikit-learn==1.6.1
 cloudpickle==2.2.1
 jsonpickle<4.0.0
 PyYAML>=6.0.1

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
+    - numpy>=2.0.0,<2.3.0
     - protobuf>=3.12,<5.0
     - smdebug_rulesconfig==1.0.1
     - importlib-metadata>=1.4.0,<7.0
@@ -64,7 +64,7 @@ dependencies:
             - multiprocess>=0.70.14
             - networkx>=3.1
             - packaging>=23.1
-            - pandas>=1.5.3
+            - pandas==2.2.3
             - pathos>=0.3.0
             - pillow>=9.5.0
             - platformdirs>=3.2.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

@@ -2,23 +2,23 @@ channels:
 - conda-forge
 dependencies:
 - python=3.10.13
-- pip<=23.3.1
+- pip<=24.3
 - pip:
-  - mlflow==2.10.2
+  - mlflow>=2.16.1
   - astunparse==1.6.3
   - cffi==1.16.0
   - cloudpickle==2.2.1
   - defusedxml==0.7.1
   - dill==0.3.9
   - gmpy2==2.1.2
-  - numpy==1.26.4
+  - 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
   - torchvision>=0.17.0
   - tqdm==4.66.2
-  - scikit-learn==1.3.2
+  - scikit-learn==1.6.1
 name: mlflow-env

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

@@ -5,10 +5,10 @@ cloudpickle==2.2.1
 defusedxml==0.7.1
 dill==0.3.9
 gmpy2==2.1.2
-numpy==1.26.4
+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

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

@@ -2,10 +2,10 @@ channels:
 - conda-forge
 dependencies:
 - python=3.10.13
-- pip<=23.3.1
+- pip<=24.3
 - pip:
-  - mlflow==2.11.1
+  - mlflow>=2.16.1
   - cloudpickle==2.2.1
-  - numpy==1.26.4
-  - tensorflow==2.16.1
+  - numpy>=1.26.4
+  - tensorflow==2.18.0
 name: mlflow-env

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

@@ -1,4 +1,4 @@
 mlflow==2.20.3
 cloudpickle==2.2.1
-numpy==1.26.4
-tensorflow==2.16.1
+numpy>=1.26.4
+tensorflow==2.18.0

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

@@ -2,14 +2,14 @@ channels:
 - conda-forge
 dependencies:
 - python=3.10.13
-- pip<=23.3.1
+- pip<=24.3
 - pip:
-  - mlflow==2.11.1
+  - mlflow>=2.16.1
   - lz4==4.3.2
-  - numpy==1.26.4
-  - pandas==2.2.1
+  - numpy>=2.0.0
+  - pandas==2.2.3
   - psutil==5.9.8
-  - scikit-learn==1.3.2
-  - scipy==1.11.3
+  - scikit-learn==1.6.1
+  - scipy==1.13.0
   - xgboost==1.7.1
 name: mlflow-env