yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-10-31 10:39:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed sagemaker either (#15156)

Browse Source
This commit is contained in:
Onkar Ravgan 2024-02-13 10:17:30 +05:30 committed by GitHub
parent 75eb54a0c8
commit 3cc074ba5a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -24,6 +24,9 @@ from metadata.generated.schema.entity.data.mlmodel import (
from metadata.generated.schema.entity.services.connections.mlmodel.sageMakerConnection import ( from metadata.generated.schema.entity.services.connections.mlmodel.sageMakerConnection import (
SageMakerConnection, SageMakerConnection,
) )
from metadata.generated.schema.entity.services.ingestionPipelines.status import (
StackTraceError,
)
from metadata.generated.schema.metadataIngestion.workflow import ( from metadata.generated.schema.metadataIngestion.workflow import (
Source as WorkflowSource, Source as WorkflowSource,
) )
@ -128,14 +131,23 @@ class SagemakerSource(MlModelServiceSource):
""" """
Prepare the Request model Prepare the Request model
""" """
try:
mlmodel_request = CreateMlModelRequest( mlmodel_request = CreateMlModelRequest(
name=model.name, name=model.name,
algorithm=self._get_algorithm(), # Setting this to a constant algorithm=self._get_algorithm(), # Setting this to a constant
mlStore=self._get_ml_store(model.name), mlStore=self._get_ml_store(model.name),
service=self.context.mlmodel_service, service=self.context.mlmodel_service,
) )
yield mlmodel_request yield Either(right=mlmodel_request)
self.register_record(mlmodel_request=mlmodel_request) self.register_record(mlmodel_request=mlmodel_request)
except Exception as exc: # pylint: disable=broad-except
yield Either(
left=StackTraceError(
name=model.name,
error=f"Error creating mlmodel: {exc}",
stackTrace=traceback.format_exc(),
)
)
def _get_ml_store( # pylint: disable=arguments-differ def _get_ml_store( # pylint: disable=arguments-differ
self, self,