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