don't repack model if entry_point is not given

Wang Napieralski · Wang Napieralski · commit dfa2716e9958 · 2021-06-17T11:05:36.000-07:00
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
@@ -1124,7 +1124,7 @@ def _upload_code(self, key_prefix, repack=False):
                 dependencies=self.dependencies,
             )
 
-        if repack and self.model_data is not None:
+        if repack and self.model_data is not None and self.entry_point is not None:
             bucket = self.bucket or self.sagemaker_session.default_bucket()
             repacked_model_data = "s3://" + "/".join([bucket, key_prefix, "model.tar.gz"])
 

Original file line number	Diff line number	Diff line change
`@@ -1124,7 +1124,7 @@ def _upload_code(self, key_prefix, repack=False):`
`1124`	`1124`	`dependencies=self.dependencies,`
`1125`	`1125`	`)`
`1126`	`1126`
`1127`		`- if repack and self.model_data is not None:`
	`1127`	`+ if repack and self.model_data is not None and self.entry_point is not None:`
`1128`	`1128`	`bucket = self.bucket or self.sagemaker_session.default_bucket()`
`1129`	`1129`	`repacked_model_data = "s3://" + "/".join([bucket, key_prefix, "model.tar.gz"])`
`1130`	`1130`