Fix arg.

apache · Sep 12, 2023 · fb38399 · fb38399
1 parent ee7a858
commit fb38399
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git a/sdks/python/apache_beam/dataframe/frame_base.py b/sdks/python/apache_beam/dataframe/frame_base.py
@@ -550,7 +550,7 @@ def wrapper(*args, **kwargs):
     f"**{BEAM_SPECIFIC!r}** for details.")
 
 
-def with_docs_from(base_type: object, name=None, removed_method=False):
+def with_docs_from(base_type, name=None, removed_method=False):
   """Decorator that updates the documentation from the wrapped function to
   duplicate the documentation from the identically-named method in `base_type`.
 

diff --git a/sdks/python/apache_beam/dataframe/frames_test.py b/sdks/python/apache_beam/dataframe/frames_test.py
@@ -813,13 +813,6 @@ def test_append_sort(self):
     self._run_test(lambda df1, df2: df2.append(df1, sort=True), df1, df2)
     self._run_test(lambda df1, df2: df2.append(df1, sort=False), df1, df2)
 
-  @unittest.skipIf(PD_VERSION < (2, 0), 'append removed in Pandas 2.0')
-  def test_append_raises_pandas_2(self):
-    df = pd.DataFrame({'A': [1, 1]})
-
-    with self.assertRaises(NotImplementedError):
-      df.append(df)
-
   def test_smallest_largest(self):
     df = pd.DataFrame({'A': [1, 1, 2, 2], 'B': [2, 3, 5, 7]})
     self._run_test(lambda df: df.nlargest(1, 'A', keep='all'), df)