diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt
index 3e6200ee3e..bfb8474e80 100644
--- a/requirements/extras/test_requirements.txt
+++ b/requirements/extras/test_requirements.txt
@@ -43,7 +43,7 @@ nbformat>=5.9,<6
 accelerate>=0.24.1,<=0.27.0
 schema==0.7.5
 tensorflow>=2.16.2,<=2.18.0
-mlflow>=2.12.2,<2.13
+mlflow>=2.19.0
 huggingface_hub==0.26.2
 uvicorn>=0.30.1
 fastapi==0.115.4
diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
index aacc85cb91..6b4290a167 100644
--- a/tests/data/serve_resources/mlflow/pytorch/requirements.txt
+++ b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
@@ -1,4 +1,4 @@
-mlflow==2.13.2
+mlflow>=2.19.0
 astunparse==1.6.3
 cffi==1.16.0
 cloudpickle==2.2.1
diff --git a/tests/data/serve_resources/mlflow/tensorflow/requirements.txt b/tests/data/serve_resources/mlflow/tensorflow/requirements.txt
index ff99d3b92e..0a0998328c 100644
--- a/tests/data/serve_resources/mlflow/tensorflow/requirements.txt
+++ b/tests/data/serve_resources/mlflow/tensorflow/requirements.txt
@@ -1,4 +1,4 @@
-mlflow==2.13.2
+mlflow>=2.19.0
 cloudpickle==2.2.1
 numpy==1.26.4
 tensorflow==2.16.1
diff --git a/tests/data/serve_resources/mlflow/xgboost/requirements.txt b/tests/data/serve_resources/mlflow/xgboost/requirements.txt
index 6f879340a7..c3dff8756a 100644
--- a/tests/data/serve_resources/mlflow/xgboost/requirements.txt
+++ b/tests/data/serve_resources/mlflow/xgboost/requirements.txt
@@ -1,4 +1,4 @@
-mlflow==2.13.2
+mlflow>=2.19.0
 lz4==4.3.2
 numpy==1.26.4
 pandas==2.0.3
diff --git a/tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py b/tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py
index 4b53c93ad4..c1c6d67a8b 100644
--- a/tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py
+++ b/tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py
@@ -48,7 +48,7 @@ def mock_mlflow_client():
 def test_encode():
     existing_names = set()
     assert encode("test-name", existing_names) == "test-name"
-    assert encode("test:name", existing_names) == "test_3a_name"
+    assert encode("test:name", existing_names) == "test:name"
     assert encode("test-name", existing_names) == "test-name_1"
 
 
@@ -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,6 +216,11 @@ 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({})
 
@@ -222,6 +228,7 @@ def getenv_side_effect(arg, default=None):
         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")