diff --git a/pymongo/mongo_client.py b/pymongo/mongo_client.py index 3de95a7f5..9bead2988 100644 --- a/pymongo/mongo_client.py +++ b/pymongo/mongo_client.py @@ -55,8 +55,7 @@ from typing import ( cast, ) -import bson -from bson.codec_options import DEFAULT_CODEC_OPTIONS, TypeRegistry +from bson.codec_options import DEFAULT_CODEC_OPTIONS, CodecOptions, TypeRegistry from bson.timestamp import Timestamp from pymongo import ( _csot, @@ -2032,7 +2031,7 @@ class MongoClient(common.BaseObject, Generic[_DocumentType]): def get_default_database( self, default: Optional[str] = None, - codec_options: Optional[bson.CodecOptions[_DocumentTypeArg]] = None, + codec_options: Optional[CodecOptions[_DocumentTypeArg]] = None, read_preference: Optional[_ServerMode] = None, write_concern: Optional[WriteConcern] = None, read_concern: Optional[ReadConcern] = None, @@ -2092,7 +2091,7 @@ class MongoClient(common.BaseObject, Generic[_DocumentType]): def get_database( self, name: Optional[str] = None, - codec_options: Optional[bson.CodecOptions[_DocumentTypeArg]] = None, + codec_options: Optional[CodecOptions[_DocumentTypeArg]] = None, read_preference: Optional[_ServerMode] = None, write_concern: Optional[WriteConcern] = None, read_concern: Optional[ReadConcern] = None,