From c1081af6bc74306175af324963cbbcfeba40f881 Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Tue, 8 Oct 2024 16:41:48 +0200
Subject: [PATCH] Fix lint

---
 libs/langgraph/langgraph/pregel/debug.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libs/langgraph/langgraph/pregel/debug.py b/libs/langgraph/langgraph/pregel/debug.py
index 985cffe75..cf90c431b 100644
--- a/libs/langgraph/langgraph/pregel/debug.py
+++ b/libs/langgraph/langgraph/pregel/debug.py
@@ -154,7 +154,7 @@ def map_debug_checkpoint(
     """Produce "checkpoint" events for stream_mode=debug."""
 
     parent_ns = config[CONF].get(CONFIG_KEY_CHECKPOINT_NS, "")
-    task_states: dict[str, RunnableConfig] = {}
+    task_states: dict[str, Union[RunnableConfig, StateSnapshot]] = {}
 
     for task in tasks:
         subgraph = None