From 73bb5da6bf3839bde6a86d1450ee93e479fef984 Mon Sep 17 00:00:00 2001 From: Supan Shah <71488507+Supan90-Shah3006@users.noreply.github.com> Date: Thu, 17 Aug 2023 01:32:55 +0530 Subject: [PATCH] ISSUE-12820: Sagemaket Metadata Success (#12843) --- .../metadata/ingestion/source/mlmodel/sagemaker/metadata.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py b/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py index f51275bf956..816e185f039 100644 --- a/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py +++ b/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py @@ -62,7 +62,7 @@ class SagemakerSource(MlModelServiceSource): def __init__(self, config: WorkflowSource, metadata_config: OpenMetadataConnection): super().__init__(config, metadata_config) - self.sagemaker = self.connection.client + self.sagemaker = self.client @classmethod def create(cls, config_dict, metadata_config: OpenMetadataConnection): @@ -91,7 +91,7 @@ class SagemakerSource(MlModelServiceSource): logger.debug(traceback.format_exc()) logger.error(f"Failed to fetch models list - {err}") - for model in models: + for model in response["Models"]: try: if filter_by_mlmodel( self.source_config.mlModelFilterPattern,