From 8571ab87e82c73ee352c81c9dc1602c585d524c9 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Wed, 28 Feb 2024 13:05:01 +0530
Subject: [PATCH] MINOR: Fix mongodb profiler imports (#15383)

---
 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 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):