Skip to content

Commit a446beb

Browse files
committed
fix: Only calculate deploy_key_prefix when needed
The `deploy_key_prefix` variable is only used conditionally in `Model.prepare_container_def`. It can be avoided when no code upload is needed.
1 parent b535ed8 commit a446beb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sagemaker/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -684,11 +684,11 @@ def prepare_container_def(
684684
Returns:
685685
dict: A container definition object usable with the CreateModel API.
686686
"""
687-
deploy_key_prefix = fw_utils.model_code_key_prefix(
688-
self.key_prefix, self.name, self.image_uri
689-
)
690687
deploy_env = copy.deepcopy(self.env)
691688
if self.source_dir or self.dependencies or self.entry_point or self.git_config:
689+
deploy_key_prefix = fw_utils.model_code_key_prefix(
690+
self.key_prefix, self.name, self.image_uri
691+
)
692692
self._upload_code(deploy_key_prefix, repack=self.is_repack())
693693
deploy_env.update(self._script_mode_env_vars())
694694

0 commit comments

Comments
 (0)