File tree 1 file changed +9
-8
lines changed
1 file changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -2825,14 +2825,15 @@ def create_model_package_from_containers(
2825
2825
drift_check_baselines = drift_check_baselines ,
2826
2826
customer_metadata_properties = customer_metadata_properties ,
2827
2827
)
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
+ )
2836
2837
return self .sagemaker_client .create_model_package (** request )
2837
2838
2838
2839
def wait_for_model_package (self , model_package_name , poll = 5 ):
You can’t perform that action at this time.
0 commit comments