diff --git a/ingestion/src/metadata/ingestion/source/looker.py b/ingestion/src/metadata/ingestion/source/looker.py index f022a0ce6c4..a05d61fb3b2 100644 --- a/ingestion/src/metadata/ingestion/source/looker.py +++ b/ingestion/src/metadata/ingestion/source/looker.py @@ -15,8 +15,8 @@ from typing import Iterable import looker_sdk -from metadata.generated.schema.entity.services.connections.dashboard import ( - lookerConnection, +from metadata.generated.schema.entity.services.connections.dashboard.lookerConnection import ( + LookerConnection, ) from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import ( OpenMetadataConnection, @@ -82,8 +82,8 @@ class LookerSource(Source[Entity]): @classmethod def create(cls, config_dict: dict, metadata_config: OpenMetadataConnection): config = WorkflowSource.parse_obj(config_dict) - connection: lookerConnection = config.serviceConnection.__root__.config - if not isinstance(connection, lookerConnection): + connection: LookerConnection = config.serviceConnection.__root__.config + if not isinstance(connection, LookerConnection): raise InvalidSourceException( f"Expected LookerConnection, but got {connection}" )