From 1dc8447a7702353c0fe69f0e4c0b0ef78d162f0d Mon Sep 17 00:00:00 2001 From: Alex Malao Date: Tue, 4 Nov 2025 11:49:00 -0500 Subject: [PATCH 1/2] added source window to assign context window mapping fn --- sdks/python/apache_beam/transforms/sideinputs.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sdks/python/apache_beam/transforms/sideinputs.py b/sdks/python/apache_beam/transforms/sideinputs.py index 7d72a02f8874..dd6482e38c2f 100644 --- a/sdks/python/apache_beam/transforms/sideinputs.py +++ b/sdks/python/apache_beam/transforms/sideinputs.py @@ -60,7 +60,8 @@ def default_window_mapping_fn( def map_via_end(source_window: window.BoundedWindow) -> window.BoundedWindow: return list( target_window_fn.assign( - window.WindowFn.AssignContext(source_window.max_timestamp())))[-1] + window.WindowFn.AssignContext(source_window.max_timestamp(), + window=source_window)))[-1] return map_via_end From a30b97ff739dcdfdf834cc4fb68b8fe8dca4f147 Mon Sep 17 00:00:00 2001 From: Alex Malao Date: Tue, 4 Nov 2025 13:34:50 -0500 Subject: [PATCH 2/2] yapf --- sdks/python/apache_beam/transforms/sideinputs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/transforms/sideinputs.py b/sdks/python/apache_beam/transforms/sideinputs.py index dd6482e38c2f..a38e05d66cbe 100644 --- a/sdks/python/apache_beam/transforms/sideinputs.py +++ b/sdks/python/apache_beam/transforms/sideinputs.py @@ -60,8 +60,8 @@ def default_window_mapping_fn( def map_via_end(source_window: window.BoundedWindow) -> window.BoundedWindow: return list( target_window_fn.assign( - window.WindowFn.AssignContext(source_window.max_timestamp(), - window=source_window)))[-1] + window.WindowFn.AssignContext( + source_window.max_timestamp(), window=source_window)))[-1] return map_via_end