sdfsdf

datahub-project · Apr 3, 2024 · 9bbccb0 · 9bbccb0
1 parent 4b7dec1
commit 9bbccb0
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/mongodb.py b/metadata-ingestion/src/datahub/ingestion/source/mongodb.py
@@ -5,7 +5,6 @@
 
 import bson.timestamp
 import pymongo.collection
-from bson import CodecOptions, DatetimeConversion
 from packaging import version
 from pydantic import PositiveInt, validator
 from pydantic.fields import Field
@@ -281,11 +280,9 @@ def __init__(self, ctx: PipelineContext, config: MongoDBConfig):
             **options,
             **self.config.options,
         }
-        codec_options: CodecOptions = CodecOptions(
-            datetime_conversion=DatetimeConversion.DATETIME_AUTO
-        )
 
-        self.mongo_client = MongoClient(self.config.connect_uri, codec_options=codec_options, **options)  # type: ignore
+        # See https://pymongo.readthedocs.io/en/stable/examples/datetimes.html#handling-out-of-range-datetimes
+        self.mongo_client = MongoClient(self.config.connect_uri, datetime_conversion="DATETIME_AUTO", **options)  # type: ignore
 
         # This cheaply tests the connection. For details, see
         # https://pymongo.readthedocs.io/en/stable/api/pymongo/mongo_client.html#pymongo.mongo_client.MongoClient