From 65847c9ce3317e198581d0a850ef8dea12b73957 Mon Sep 17 00:00:00 2001 From: ulixius9 Date: Wed, 28 Feb 2024 12:21:12 +0530 Subject: [PATCH] MINOR: Fix mongodb profiler imports --- ingestion/src/metadata/profiler/adaptors/mongodb.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/ingestion/src/metadata/profiler/adaptors/mongodb.py b/ingestion/src/metadata/profiler/adaptors/mongodb.py index 22194535bb91..60f1438b5cf8 100644 --- a/ingestion/src/metadata/profiler/adaptors/mongodb.py +++ b/ingestion/src/metadata/profiler/adaptors/mongodb.py @@ -16,17 +16,20 @@ 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):