Fix handling of new default for method arg for Pandas 2.1

apache · Oct 4, 2023 · 46c87c4 · 46c87c4
1 parent c01b41f
commit 46c87c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/dataframe/frames.py b/sdks/python/apache_beam/dataframe/frames.py
@@ -1388,7 +1388,7 @@ def align(self, other, join, axis, level, method, **kwargs):
     Only the default, ``method=None``, is allowed."""
     if level is not None:
       raise NotImplementedError('per-level align')
-    if method is not None:
+    if method is not None and method != lib.no_default:
       raise frame_base.WontImplementError(
           f"align(method={method!r}) is not supported because it is "
           "order sensitive. Only align(method=None) is supported.",
@@ -2580,7 +2580,7 @@ def align(self, other, join, axis, copy, level, method, **kwargs):
           "align(copy=False) is not supported because it might be an inplace "
           "operation depending on the data. Please prefer the default "
           "align(copy=True).")
-    if method is not None:
+    if method is not None and method != lib.no_default:
       raise frame_base.WontImplementError(
           f"align(method={method!r}) is not supported because it is "
           "order sensitive. Only align(method=None) is supported.",