resolve conflict dependency with numpy 2.0

Author: Roja Reddy Sareddy

doc/requirements.txt

@@ -1,7 +1,7 @@
 sphinx==5.1.1
 sphinx-rtd-theme==0.5.0
 docutils==0.15.2
-packaging==20.9
+packaging>=23.0,<25
 jinja2==3.1.6
 schema==0.7.5
 accelerate>=0.24.1,<=0.27.0

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

@@ -6,6 +6,6 @@ dependencies:
 - pip:
   - mlflow>=2.16.1
   - cloudpickle==2.2.1
-  - numpy
+  - 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.16.1
 cloudpickle==2.2.1
-numpy
+numpy>=1.26.4
 tensorflow==2.18.0

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

@@ -6,7 +6,7 @@ dependencies:
 - pip:
   - mlflow>=2.16.1
   - lz4==4.3.2
-  - numpy
+  - numpy>=1.26.4
   - pandas==2.2.3
   - psutil==5.9.8
   - scikit-learn==1.6.1

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

@@ -1,6 +1,6 @@
 mlflow>=2.16.1
 lz4==4.3.2
-numpy
+numpy>=1.26.4
 pandas==2.2.3
 psutil==5.9.8
 scikit-learn==1.6.1

tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py

@@ -188,6 +188,7 @@ def getenv_side_effect(arg, default=None):
             Mock(spec=requests.Response),
             Mock(spec=requests.Response),
         ],
+        "https://test.sagemaker.aws/api/2.0/mlflow/runs/update": Mock(spec=requests.Response),
         "https://test.sagemaker.aws/api/2.0/mlflow/runs/terminate": Mock(spec=requests.Response),
     }
 
@@ -215,13 +216,19 @@ def getenv_side_effect(arg, default=None):
         mock_response.status_code = 200
         mock_response.text = json.dumps({})
 
+    mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/update"].status_code = 200
+    mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/update"].text = json.dumps(
+        {"run_id": "test_run_id", "status": "FINISHED"}
+    )
+
     mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/terminate"].status_code = 200
     mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/terminate"].text = json.dumps({})
 
     mock_request.side_effect = [
         mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/experiments/get-by-name"],
         mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/create"],
         *mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/log-batch"],
+        mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/update"],
         mock_responses["https://test.sagemaker.aws/api/2.0/mlflow/runs/terminate"],
     ]
 
@@ -231,7 +238,7 @@ def getenv_side_effect(arg, default=None):
 
     log_to_mlflow(metrics, params, tags)
 
-    assert mock_request.call_count == 6  # Total number of API calls
+    assert mock_request.call_count == 7  # Total number of API calls
 
 
 @patch("sagemaker.mlflow.forward_sagemaker_metrics.get_training_job_details")