yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-26 01:15:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

ISSUE-12820: Sagemaket Metadata Success (#12843)

Browse Source
This commit is contained in:
Supan Shah 2023-08-17 01:32:55 +05:30 committed by GitHub
parent ad3871b3bc
commit 73bb5da6bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ingestion/src/metadata/ingestion/source/mlmodel/sagemaker/metadata.py
View File

@ -62,7 +62,7 @@ class SagemakerSource(MlModelServiceSource):
def __init__(self, config: WorkflowSource, metadata_config: OpenMetadataConnection): def __init__(self, config: WorkflowSource, metadata_config: OpenMetadataConnection):
super().__init__(config, metadata_config) super().__init__(config, metadata_config)
self.sagemaker = self.connection.client self.sagemaker = self.client
@classmethod @classmethod
def create(cls, config_dict, metadata_config: OpenMetadataConnection): def create(cls, config_dict, metadata_config: OpenMetadataConnection):
@ -91,7 +91,7 @@ class SagemakerSource(MlModelServiceSource):
logger.debug(traceback.format_exc()) logger.debug(traceback.format_exc())
logger.error(f"Failed to fetch models list - {err}") logger.error(f"Failed to fetch models list - {err}")
for model in models: for model in response["Models"]:
try: try:
if filter_by_mlmodel( if filter_by_mlmodel(
self.source_config.mlModelFilterPattern, self.source_config.mlModelFilterPattern,