Skip to content

Commit d0f04cc

Browse files
committed
refactor(workflow): Remove unused node_name_map parameter from simplify_event_with_node
Removed node_name_map from simplify_event_with_node signature and its call sites in workflow_testing_utils.py as it was no longer used. Change-Id: I1ffb506ecd1c50e2c64e3575440e32b45bf41965
1 parent 39b16a9 commit d0f04cc

1 file changed

Lines changed: 2 additions & 9 deletions

File tree

tests/unittests/workflow/workflow_testing_utils.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -198,11 +198,8 @@ def _build_node_name_map(events: list[Event]) -> dict[str, str]:
198198

199199
def simplify_event_with_node(
200200
event: Event,
201-
node_name_map: dict[str, str] | None = None,
202201
include_state_delta: bool = False,
203202
) -> Any | None:
204-
if node_name_map is None:
205-
node_name_map = {}
206203
if isinstance(event, Event):
207204
if (
208205
'output' not in event.model_fields_set
@@ -265,9 +262,7 @@ def simplify_events_with_node(
265262
):
266263
continue
267264

268-
simplified_event = simplify_event_with_node(
269-
event, node_name_map, include_state_delta
270-
)
265+
simplified_event = simplify_event_with_node(event, include_state_delta)
271266
if simplified_event:
272267
# Map the author to the source node name if it exists.
273268
if hasattr(event, 'node_info') and event.node_info.path:
@@ -312,9 +307,7 @@ def simplify_events_with_node_and_agent_state(
312307
and '/' not in (event.node_info.path or '')
313308
):
314309
continue
315-
simplified_event = simplify_event_with_node(
316-
event, node_name_map, include_state_delta
317-
)
310+
simplified_event = simplify_event_with_node(event, include_state_delta)
318311

319312
# Map the author to the source node name if it exists.
320313
if hasattr(event, 'node_info') and event.node_info.path:

0 commit comments

Comments
 (0)