Fix/pymongo new version

CHANGELOG.md

@@ -30,6 +30,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Fixed default event.uuid
 - Fixed atlas vector search
 - Fix the bug where shared artifacts are deleted when removing a component.
+- Fix compatibility issues with the latest version of pymongo.
 
 #### New Features & Functionality
 

plugins/mongodb/superduper_mongodb/__init__.py

@@ -3,7 +3,7 @@
 from .metadata import MongoMetaDataStore as MetaDataStore
 from .query import MongoQuery
 
-__version__ = "0.0.4"
+__version__ = "0.0.5"
 
 __all__ = [
     "ArtifactStore",

plugins/mongodb/superduper_mongodb/data_backend.py

@@ -4,6 +4,7 @@
 import click
 import mongomock
 import pymongo
+import pymongo.collection
 from superduper import CFG, logging
 from superduper.backends.base.data_backend import BaseDataBackend
 from superduper.backends.base.metadata import MetaDataStoreProxy

superduper/base/variables.py

@@ -23,7 +23,11 @@ def _replace_variables(x, **kwargs):
             _replace_variables(k, **kwargs): _replace_variables(v, **kwargs)
             for k, v in x.items()
         }
-    if isinstance(x, str) and re.match(r'^<var:(.*?)>$', x) is not None and '<' not in x[1:-1]:
+    if (
+        isinstance(x, str)
+        and re.match(r'^<var:(.*?)>$', x) is not None
+        and '<' not in x[1:-1]
+    ):
         return kwargs.get(x[5:-1], x)
     if isinstance(x, str):
         variables = re.findall(r'<var:(.*?)>', x)

superduper/components/template.py

@@ -23,6 +23,7 @@ class _BaseTemplate(Component):
     :param blobs: Blob identifiers in `Template.component`.
     :param files: File identifiers in `Template.component`.
     :param substitutions: Substitutions to be made to create variables.
+    :param default_values: Default values for the variables.
     """
 
     literals: t.ClassVar[t.Tuple[str]] = ('template',)
@@ -61,7 +62,11 @@ def form_template(self):
         return {
             'identifier': '<enter-a-unique-identifier>',
             '_variables': {
-                k: (f'<value-{i}>' if k not in self.default_values else self.default_values[k])
+                k: (
+                    f'<value-{i}>'
+                    if k not in self.default_values
+                    else self.default_values[k]
+                )
                 for i, k in enumerate(self.template_variables)
             },
             **{k: v for k, v in self.template.items() if k != 'identifier'},
@@ -187,7 +192,11 @@ def form_template(self):
         """Form to be diplayed to user."""
         return {
             '_variables': {
-                k: (f'<value-{i}>' if k not in self.default_values else self.default_values[k])
+                k: (
+                    f'<value-{i}>'
+                    if k not in self.default_values
+                    else self.default_values[k]
+                )
                 for i, k in enumerate(self.template_variables)
             },
             **{