Update describe_mlflow_tracking_server call to align with api changes (aws#1466)

jiapinw · knikure · commit ca40ac81b1f7 · 2024-06-12T20:48:10.000Z
diff --git a/src/sagemaker/serve/utils/lineage_utils.py b/src/sagemaker/serve/utils/lineage_utils.py
@@ -215,7 +215,7 @@ def _retrieve_and_create_if_not_exist_mlflow_model_path_lineage_artifact(
         match = re.match(TRACKING_SERVER_ARN_REGEX, tracking_server_arn)
         mlflow_tracking_server_name = match.group(4)
         describe_result = sagemaker_session.sagemaker_client.describe_mlflow_tracking_server(
-            MlflowTrackingServerName=mlflow_tracking_server_name
+            TrackingServerName=mlflow_tracking_server_name
         )
         tracking_server_creation_time = describe_result["CreationTime"].strftime(
             TRACKING_SERVER_CREATION_TIME_FORMAT

Original file line number	Diff line number	Diff line change
`@@ -215,7 +215,7 @@ def _retrieve_and_create_if_not_exist_mlflow_model_path_lineage_artifact(`
`215`	`215`	`match = re.match(TRACKING_SERVER_ARN_REGEX, tracking_server_arn)`
`216`	`216`	`mlflow_tracking_server_name = match.group(4)`
`217`	`217`	`describe_result = sagemaker_session.sagemaker_client.describe_mlflow_tracking_server(`
`218`		`- MlflowTrackingServerName=mlflow_tracking_server_name`
	`218`	`+ TrackingServerName=mlflow_tracking_server_name`
`219`	`219`	`)`
`220`	`220`	`tracking_server_creation_time = describe_result["CreationTime"].strftime(`
`221`	`221`	`TRACKING_SERVER_CREATION_TIME_FORMAT`