fix: Add kms key support for Model registration (#3784)

mufaddal-rohawala · web-flow · commit 4884d1843024 · 2023-04-26T17:11:32.000-07:00
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
@@ -574,6 +574,7 @@ def _upload_code(self, key_prefix: str, repack: bool = False) -> None:
                 script=self.entry_point,
                 directory=self.source_dir,
                 dependencies=self.dependencies,
+                kms_key=self.model_kms_key,
                 settings=self.sagemaker_session.settings,
             )
 

Original file line number	Diff line number	Diff line change
`@@ -574,6 +574,7 @@ def _upload_code(self, key_prefix: str, repack: bool = False) -> None:`
`574`	`574`	`script=self.entry_point,`
`575`	`575`	`directory=self.source_dir,`
`576`	`576`	`dependencies=self.dependencies,`
	`577`	`+ kms_key=self.model_kms_key,`
`577`	`578`	`settings=self.sagemaker_session.settings,`
`578`	`579`	`)`
`579`	`580`