resolve conflict dependency with numpy 2.0

Author: Roja Reddy Sareddy

Diff for: tests/data/serve_resources/mlflow/pytorch/conda.yaml

@@ -4,7 +4,7 @@ dependencies:
 - python=3.10.13
 - pip<=23.3.1
 - pip:
-  - mlflow==2.16.1
+  - mlflow>=2.16.1
   - astunparse==1.6.3
   - cffi==1.16.0
   - cloudpickle==2.2.1

Diff for: tests/data/serve_resources/mlflow/pytorch/requirements.txt

@@ -1,4 +1,4 @@
-mlflow==2.16.1
+mlflow>=2.16.1
 astunparse==1.6.3
 cffi==1.16.0
 cloudpickle==2.2.1

Diff for: tests/data/serve_resources/mlflow/tensorflow/conda.yaml

@@ -4,8 +4,8 @@ dependencies:
 - python=3.10.13
 - pip<=23.3.1
 - pip:
-  - mlflow==2.16.1
+  - mlflow>=2.16.1
   - cloudpickle==2.2.1
-  - numpy>=2.0.0,<2.3.0
-  - tensorflow==2.17.0
+  - numpy==2.0.0
+  - tensorflow==2.18.0
 name: mlflow-env

Diff for: tests/data/serve_resources/mlflow/tensorflow/requirements.txt

@@ -1,4 +1,4 @@
-mlflow==2.16.1
+mlflow>=2.16.1
 cloudpickle==2.2.1
-numpy>=2.0.0,<2.3.0
-tensorflow==2.17.0
+numpy==2.0.0
+tensorflow==2.18.0

Diff for: tests/data/serve_resources/mlflow/xgboost/conda.yaml

@@ -4,12 +4,12 @@ dependencies:
 - python=3.10.13
 - pip<=23.3.1
 - pip:
-  - mlflow==2.16.1
+  - mlflow>=2.16.1
   - lz4==4.3.2
   - numpy>=2.0.0,<2.3.0
   - pandas==2.2.3
   - psutil==5.9.8
   - scikit-learn==1.6.1
   - scipy==1.13.0
-  - xgboost==1.7.1
+  - xgboost==2.1.4
 name: mlflow-env

Diff for: tests/data/serve_resources/mlflow/xgboost/requirements.txt

@@ -1,8 +1,8 @@
-mlflow==2.16.1
+mlflow>=2.16.1
 lz4==4.3.2
 numpy>=2.0.0,<2.3.0
 pandas==2.2.3
 psutil==5.9.8
 scikit-learn==1.6.1
 scipy==1.13.0
-xgboost==1.7.1
+xgboost==2.1.4

Diff for: tests/unit/sagemaker/serve/detector/test_dependency_manager.py

@@ -34,7 +34,7 @@
 
 EXPECTED_DEPENDENCY_MAP = {
     "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",