diff --git a/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py b/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py index 7ab598f6d68..f5aad46069d 100644 --- a/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py +++ b/ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py @@ -98,14 +98,14 @@ class SagemakerSource(MlModelServiceSource): try: while has_more_models: response = self.sagemaker.list_models(**args) - models.append(response["Models"]) + models.extend(response["Models"]) has_more_models = response.get("NextToken") args["NextToken"] = response.get("NextToken") except Exception as err: logger.debug(traceback.format_exc()) logger.error(f"Failed to fetch models list - {err}") - for model in response["Models"]: + for model in models: try: if filter_by_mlmodel( self.source_config.mlModelFilterPattern,