MINOR: Fix mongodb profiler imports (#15383)

This commit is contained in:
Mayur Singal 2024-02-28 13:05:01 +05:30 committed by GitHub
parent 7962524619
commit 8571ab87e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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