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
934
934
self .built_model = self ._build_for_model_server ()
935
935
return self .built_model
936
936
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
+
937
946
if isinstance (self .model , str ):
938
947
model_task = None
939
948
@@ -992,6 +1001,7 @@ def build( # pylint: disable=R0911
992
1001
993
1002
raise ValueError ("%s model server is not supported" % self .model_server )
994
1003
1004
+
995
1005
def _build_validations (self ):
996
1006
"""Validations needed for model server overrides, or auto-detection or fallback"""
997
1007
if self .inference_spec and self .model :
You can’t perform that action at this time.
0 commit comments