MINOR: Mariadb Lineage Fix (#19575)

This commit is contained in:
Mayur Singal 2025-01-29 10:59:16 +05:30 committed by GitHub
parent 8c5eee8f07
commit 39eae2dfa5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -14,7 +14,7 @@ Mariadb lineage module
from typing import Optional from typing import Optional
from metadata.generated.schema.entity.services.connections.database.mariaDBConnection import ( from metadata.generated.schema.entity.services.connections.database.mariaDBConnection import (
MariadbConnection, MariaDBConnection,
) )
from metadata.generated.schema.metadataIngestion.workflow import ( from metadata.generated.schema.metadataIngestion.workflow import (
Source as WorkflowSource, Source as WorkflowSource,
@ -38,8 +38,8 @@ class MariadbLineageSource(LineageSource):
): ):
"""Create class instance""" """Create class instance"""
config: WorkflowSource = WorkflowSource.model_validate(config_dict) config: WorkflowSource = WorkflowSource.model_validate(config_dict)
connection: MariadbConnection = config.serviceConnection.root.config connection: MariaDBConnection = config.serviceConnection.root.config
if not isinstance(connection, MariadbConnection): if not isinstance(connection, MariaDBConnection):
raise InvalidSourceException( raise InvalidSourceException(
f"Expected MariadbConnection, but got {connection}" f"Expected MariadbConnection, but got {connection}"
) )