Remove deprecated week and weekofyear for Pandas 2 (#28492)

apache · Sep 20, 2023 · a7335cb · a7335cb
1 parent 93c3224
commit a7335cb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/dataframe/frames.py b/sdks/python/apache_beam/dataframe/frames.py
@@ -5377,11 +5377,12 @@ def func(df, *args, **kwargs):
   'second',
   'time',
   'timetz',
-  'week',
   'weekday',
-  'weekofyear',
   'year',
 ]
+# Pandas 2 removed these.
+if PD_VERSION < (2, 0):
+  ELEMENTWISE_DATETIME_PROPERTIES += ['week', 'weekofyear']
 
 for method in ELEMENTWISE_DATETIME_PROPERTIES:
   setattr(_DeferredDatetimeMethods,