in aws-sagemaker-modelpackagegroup/src/main/java/software/amazon/sagemaker/modelpackagegroup/CreateHandler.java [97:127]
private boolean stabilizedOnCreate(
final CreateModelPackageGroupRequest createModelPackageGroupRequest,
final CreateModelPackageGroupResponse createModelPackageGroupResponse,
final ProxyClient<SageMakerClient> proxyClient,
final ResourceModel model,
final CallbackContext callbackContext) {
if(model.getModelPackageGroupArn() == null){
model.setModelPackageGroupArn(createModelPackageGroupResponse.modelPackageGroupArn());
}
final DescribeModelPackageGroupResponse response = proxyClient.injectCredentialsAndInvokeV2(Translator.translateToReadRequest(model),
proxyClient.client()::describeModelPackageGroup);
final ModelPackageGroupStatus modelPackageGroupStatus = response.modelPackageGroupStatus();
switch (modelPackageGroupStatus) {
case COMPLETED:
case FAILED:
logger.log(String.format("%s [%s] has been stabilized with status %s.", ResourceModel.TYPE_NAME,
model.getPrimaryIdentifier(), modelPackageGroupStatus));
return true;
case PENDING:
case IN_PROGRESS:
logger.log(String.format("%s [%s] is stabilizing.", ResourceModel.TYPE_NAME, model.getPrimaryIdentifier()));
return false;
default:
throw new CfnGeneralServiceException(
"Stabilizing of " + model.getPrimaryIdentifier()
+ " failed with unexpected status " + modelPackageGroupStatus);
}
}