diff --git a/ingestion/src/metadata/profiler/adaptors/mongodb.py b/ingestion/src/metadata/profiler/adaptors/mongodb.py index 22194535bb9..60f1438b5cf 100644 --- a/ingestion/src/metadata/profiler/adaptors/mongodb.py +++ b/ingestion/src/metadata/profiler/adaptors/mongodb.py @@ -16,17 +16,20 @@ from enum import Enum from typing import TYPE_CHECKING, Dict, List, Optional, Union from pydantic import BaseModel, Field -from pymongo.command_cursor import CommandCursor -from pymongo.cursor import Cursor from metadata.generated.schema.entity.data.table import Column, Table from metadata.profiler.adaptors.nosql_adaptor import NoSQLAdaptor from metadata.utils.sqa_like_column import SQALikeColumn +# pylint: disable=invalid-name if TYPE_CHECKING: from pymongo import MongoClient + from pymongo.command_cursor import CommandCursor + from pymongo.cursor import Cursor else: - MongoClient = None # pylint: disable=invalid-name + MongoClient = None + CommandCursor = None + Cursor = None class AggregationFunction(Enum):