resolve unit tests for mlflow

Author: Roja Reddy Sareddy

tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py

@@ -217,10 +217,9 @@ def getenv_side_effect(arg, default=None):
         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/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({})