Skip to content

Commit cda6b3c

Browse files
committed
Fixing black-check errors
1 parent a1182f1 commit cda6b3c

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

src/sagemaker/session.py

+9-8
Original file line numberDiff line numberDiff line change
@@ -2825,14 +2825,15 @@ def create_model_package_from_containers(
28252825
drift_check_baselines=drift_check_baselines,
28262826
customer_metadata_properties=customer_metadata_properties,
28272827
)
2828-
try:
2829-
self.sagemaker_client.describe_model_package_group(
2830-
ModelPackageGroupName=request["ModelPackageGroupName"]
2831-
)
2832-
except ClientError:
2833-
self.sagemaker_client.create_model_package_group(
2834-
ModelPackageGroupName=request["ModelPackageGroupName"]
2835-
)
2828+
if model_package_group_name is not None:
2829+
try:
2830+
self.sagemaker_client.describe_model_package_group(
2831+
ModelPackageGroupName=request["ModelPackageGroupName"]
2832+
)
2833+
except ClientError:
2834+
self.sagemaker_client.create_model_package_group(
2835+
ModelPackageGroupName=request["ModelPackageGroupName"]
2836+
)
28362837
return self.sagemaker_client.create_model_package(**request)
28372838

28382839
def wait_for_model_package(self, model_package_name, poll=5):

0 commit comments

Comments
 (0)