Skip to content

Commit 925c980

Browse files
authored
Raise error early for MLFlow in local container mode
1 parent a58654e commit 925c980

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -934,6 +934,15 @@ def build( # pylint: disable=R0911
934934
self.built_model = self._build_for_model_server()
935935
return self.built_model
936936

937+
if self.mode == Mode.LOCAL_CONTAINER and (
938+
self.model_metadata and (
939+
"MLFLOW_MODEL_PATH" in self.model_metadata or
940+
"MLFLOW_TRACKING_ARN" in self.model_metadata
941+
)
942+
):
943+
raise ValueError("MLflow model support is not available in Local Container mode. "
944+
"Please use SageMaker Endpoint mode for MLflow models.")
945+
937946
if isinstance(self.model, str):
938947
model_task = None
939948

@@ -992,6 +1001,7 @@ def build( # pylint: disable=R0911
9921001

9931002
raise ValueError("%s model server is not supported" % self.model_server)
9941003

1004+
9951005
def _build_validations(self):
9961006
"""Validations needed for model server overrides, or auto-detection or fallback"""
9971007
if self.inference_spec and self.model:

0 commit comments

Comments
 (0)