NUmpy FIxes

Author: nargokul

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.9.0,<2.0
+    - numpy>=1.9.0,<3.0
     - protobuf>=3.12,<5.0
     - smdebug_rulesconfig==1.0.1
     - importlib-metadata>=1.4.0,<7.0

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

@@ -5,7 +5,7 @@ cloudpickle==2.2.1
 defusedxml==0.7.1
 dill==0.3.8
 gmpy2==2.1.2
-numpy==1.24.4
+numpy>=1.24.4
 opt-einsum==3.3.0
 packaging==21.3
 pandas==2.2.1

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

@@ -6,6 +6,6 @@ dependencies:
 - pip:
   - mlflow==2.11.1
   - cloudpickle==2.2.1
-  - numpy==1.26.4
+  - numpy>=1.26.4
   - tensorflow==2.16.1
 name: mlflow-env

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

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

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

@@ -6,7 +6,7 @@ dependencies:
 - pip:
   - mlflow==2.11.1
   - lz4==4.3.2
-  - numpy==1.26.4
+  - numpy>=1.26.4
   - pandas==2.2.1
   - psutil==5.9.8
   - scikit-learn==1.3.2

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

@@ -1,6 +1,6 @@
 mlflow==2.13.2
 lz4==4.3.2
-numpy==1.24.4
+numpy>=1.24.4
 pandas==2.0.3
 psutil==5.9.8
 scikit-learn==1.3.2