(beam) branch master updated: Allow passing label through GBK (#36426)

[tv...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tvalentyn pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new 27ad139bb18 Allow passing label through GBK (#36426)
27ad139bb18 is described below

commit 27ad139bb1886f7670017d26cf487254f13e7327
Author: Danny McCormick <da...@google.com>
AuthorDate: Tue Oct 7 18:06:22 2025 -0400

    Allow passing label through GBK (#36426)
---
 sdks/python/apache_beam/transforms/core.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sdks/python/apache_beam/transforms/core.py b/sdks/python/apache_beam/transforms/core.py
index 3aae9f08344..2126169a57f 100644
--- a/sdks/python/apache_beam/transforms/core.py
+++ b/sdks/python/apache_beam/transforms/core.py
@@ -3335,9 +3335,10 @@ class GroupByKey(PTransform):
 
   The implementation here is used only when run on the local direct runner.
   """
-  def __init__(self):
+  def __init__(self, label=None):
     self._replaced_by_gbek = False
     self._inside_gbek = False
+    super().__init__(label)
 
   class ReifyWindows(DoFn):
     def process(