Skip to content

Commit dcaef38

Browse files
committed
change: simplify _upload_code invocation
1 parent 7295190 commit dcaef38

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/sagemaker/model.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -399,12 +399,12 @@ def prepare_container_def(
399399
)
400400
deploy_env = copy.deepcopy(self.env)
401401
if self.source_dir or self.dependencies or self.entry_point or self.git_config:
402-
if self.key_prefix or self.git_config:
403-
self._upload_code(deploy_key_prefix, repack=False)
404-
elif self.source_dir and self.entry_point:
405-
self._upload_code(deploy_key_prefix, repack=True)
406-
else:
407-
self._upload_code(deploy_key_prefix, repack=False)
402+
self._upload_code(
403+
deploy_key_prefix,
404+
repack=self.source_dir
405+
and self.entry_point
406+
and not (self.key_prefix or self.git_config),
407+
)
408408
deploy_env.update(self._script_mode_env_vars())
409409
return sagemaker.container_def(
410410
self.image_uri, self.model_data, deploy_env, image_config=self.image_config

0 commit comments

Comments
 (0)