Fix

langchain-ai · Nov 17, 2024 · 5adb3e4 · 5adb3e4
1 parent 66f97a9
commit 5adb3e4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/libs/checkpoint/langgraph/checkpoint/memory/__init__.py b/libs/checkpoint/langgraph/checkpoint/memory/__init__.py
@@ -77,11 +77,9 @@ def __init__(
         self.storage = factory(lambda: defaultdict(dict))
         self.writes = factory(dict)
         self.stack = ExitStack()
-        try:
+        if factory is not defaultdict:
             self.stack.enter_context(self.storage)  # type: ignore[arg-type]
             self.stack.enter_context(self.writes)  # type: ignore[arg-type]
-        except (TypeError, AttributeError):
-            pass
 
     def __enter__(self) -> "MemorySaver":
         return self.stack.__enter__()