Skip to content

Commit 375c0fd

Browse files
fix: cr comments
1 parent d6d8e08 commit 375c0fd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/estimator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1007,7 +1007,8 @@ def register(
10071007
if compile_model_family is not None:
10081008
model = self._compiled_models[compile_model_family]
10091009
else:
1010-
kwargs["model_kms_key"] = self.output_kms_key
1010+
if "model_kms_key" not in kwargs:
1011+
kwargs["model_kms_key"] = self.output_kms_key
10111012
model = self.create_model(image_uri=image_uri, **kwargs)
10121013
model.name = model_name
10131014
return model.register(

0 commit comments

Comments
 (0)