def get_data_check_baseline_uri()

in batch_pipeline/infra/model_registry.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

    def get_data_check_baseline_uri(self, model_package_arn: str):
        try:
            model_details = self.sm_client.describe_model_package(ModelPackageName=model_package_arn)
            print(model_details)
            baseline_uri = model_details['DriftCheckBaselines']['ModelDataQuality']['Constraints']['S3Uri']
            baseline_uri = baseline_uri.replace('/constraints.json','') # returning the folder containing constraints and statistics
            return baseline_uri
        except ClientError as e:
            error_message = e.response["Error"]["Message"]
            logger.error(error_message)
            raise Exception(error_message)