Skip to content

Commit 09aa2e1

Browse files
fix: maketplace integs (#3925)
1 parent ecbffbf commit 09aa2e1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1774,6 +1774,7 @@ def _create_sagemaker_model(self, *args, **kwargs): # pylint: disable=unused-ar
17741774
self.sagemaker_session.wait_for_model_package(model_package_name)
17751775
self._created_model_package_name = model_package_name
17761776
model_package_name = self._created_model_package_name
1777+
container_def = {"ModelPackageName": model_package_name}
17771778
else:
17781779
# When a ModelPackageArn is provided we just create the Model
17791780
match = re.match(MODEL_PACKAGE_ARN_PATTERN, self.model_package_arn)
@@ -1782,8 +1783,7 @@ def _create_sagemaker_model(self, *args, **kwargs): # pylint: disable=unused-ar
17821783
else:
17831784
# model_package_arn can be just the name if your account owns the Model Package
17841785
model_package_name = self.model_package_arn
1785-
1786-
container_def = {"ModelPackageName": self.model_package_arn}
1786+
container_def = {"ModelPackageName": self.model_package_arn}
17871787

17881788
if self.env != {}:
17891789
container_def["Environment"] = self.env

0 commit comments

Comments
 (0)