We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a58654e commit f74fd75Copy full SHA for f74fd75
src/sagemaker/serve/builder/model_builder.py
@@ -883,6 +883,16 @@ def build( # pylint: disable=R0911
883
884
if mode:
885
self.mode = mode
886
+
887
+ if self.mode == Mode.LOCAL_CONTAINER and (
888
+ self.model_metadata and (
889
+ "MLFLOW_MODEL_PATH" in self.model_metadata or
890
+ "MLFLOW_TRACKING_ARN" in self.model_metadata
891
+ )
892
+ ):
893
+ raise ValueError("MLflow model support is not available in Local Container mode. "
894
+ "Please use SageMaker Endpoint mode for MLflow models.")
895
896
if role_arn:
897
self.role_arn = role_arn
898
0 commit comments