File tree Expand file tree Collapse file tree 1 file changed +10
-0
lines changed
src/sagemaker/serve/builder Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments