Skip to content

Commit ef346b1

Browse files
author
Changli Han
committed
add code_location to create_model
1 parent e163807 commit ef346b1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/sagemaker/estimator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1664,6 +1664,7 @@ def deploy(
16641664
model = self._compiled_models[family]
16651665
else:
16661666
kwargs["model_kms_key"] = self.output_kms_key
1667+
kwargs["code_location"] = self.code_location
16671668
model = self.create_model(**kwargs)
16681669

16691670
model.name = model_name

0 commit comments

Comments
 (0)