Skip to content

Commit 804123c

Browse files
committed
Improve error message
1 parent f74fd75 commit 804123c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -890,8 +890,8 @@ def build( # pylint: disable=R0911
890890
"MLFLOW_TRACKING_ARN" in self.model_metadata
891891
)
892892
):
893-
raise ValueError("MLflow model support is not available in Local Container mode. "
894-
"Please use SageMaker Endpoint mode for MLflow models.")
893+
raise ValueError("MLflow model support is not available for Mode.LOCAL_CONTAINER. "
894+
"Please use Mode.SAGEMAKER_ENDPOINT for MLflow models.")
895895

896896
if role_arn:
897897
self.role_arn = role_arn

0 commit comments

Comments
 (0)