Only populate kwargs if not there

apache · Sep 12, 2023 · ee7a858 · ee7a858
1 parent 944770e
commit ee7a858
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/dataframe/frame_base.py b/sdks/python/apache_beam/dataframe/frame_base.py
@@ -520,7 +520,8 @@ def wrapper(*args, **kwargs):
       # Still have to populate these for the Beam function signature.
       if removed_args:
         for name in removed_args:
-          kwargs[name] = None
+          if not name in kwargs:
+            kwargs[name] = None
       return func(**kwargs)
 
     return wrapper