typing fixes

Author: NoahStapp

pymongo/message.py

@@ -1119,7 +1119,7 @@ def _check_doc_size_limits(
         if real_op_type == "insert":
             doc = op_doc["document"]
             if not isinstance(doc, RawBSONDocument):
-                doc_to_encode = op_doc.copy()  # Shallow copy
+                doc_to_encode = op_doc.copy()  # type: ignore[attr-defined] # Shallow copy
                 doc_to_encode["document"] = ChainMap(doc, {"_id": doc["_id"]})  # type: ignore[index]
 
         # Encode current operation doc and, if newly added, namespace doc.

test/asynchronous/test_client_bulk_write.py

@@ -18,7 +18,7 @@
 import os
 import sys
 
-from bson import encode
+from bson import RawBSONDocument, encode
 
 sys.path[0:0] = [""]
 

test/test_client_bulk_write.py

@@ -18,7 +18,7 @@
 import os
 import sys
 
-from bson import encode
+from bson import RawBSONDocument, encode
 
 sys.path[0:0] = [""]