MLX-1224 add code_location to create_model

aws · Mar 25, 2024 · 2f5345a · 2f5345a
1 parent e163807
commit 2f5345a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -1664,6 +1664,7 @@ def deploy(
             model = self._compiled_models[family]
         else:
             kwargs["model_kms_key"] = self.output_kms_key
+            kwargs["code_location"] = self.code_location
             model = self.create_model(**kwargs)
 
         model.name = model_name