fix tests

materialsproject · May 29, 2023 · 589624d · 589624d
1 parent d8f99c9
commit 589624d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/maggma/stores/mongolike.py b/src/maggma/stores/mongolike.py
@@ -704,8 +704,8 @@ def __init__(
         self,
         paths: Union[str, List[str]],
         read_only: bool = True,
-        serialization_option: int | None = None,
-        serialization_default: Any | None = None,
+        serialization_option: Optional[int] = None,
+        serialization_default: Any = None,
         **kwargs,
     ):
         """

diff --git a/tests/stores/test_mongolike.py b/tests/stores/test_mongolike.py
@@ -518,7 +518,7 @@ class SubFloat(float):
             "a.json",
             read_only=False,
             serialization_option=None,
-            serialization_default="test",
+            serialization_default=lambda x: "test",
         )
         jsonstore.connect()
         jsonstore.update({"wrong_field": SubFloat(1.1), "task_id": 3})