From 73f9ef0ef873a3c72080557b2c639cced514a58b Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Wed, 19 Mar 2025 16:00:49 -0400 Subject: [PATCH] add type --- libs/langgraph/langgraph/pregel/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 661262071..256ddbd6c 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -5,7 +5,7 @@ import concurrent import concurrent.futures import queue import weakref -from collections import deque +from collections import defaultdict, deque from functools import partial from typing import ( Any, @@ -2753,7 +2753,7 @@ class Pregel(PregelProtocol): def _trigger_to_nodes(nodes: dict[str, PregelNode]) -> dict[str, list[str]]: """Index from a trigger to nodes that depend on it.""" - trigger_to_nodes = {} + trigger_to_nodes: defaultdict[str, list[str]] = defaultdict(list) for name, node in nodes.items(): for trigger in node.triggers: trigger_to_nodes.setdefault(trigger, []).append(name)