From fb38399cda8d7c7bd8aac77a5d6faba51e64fa10 Mon Sep 17 00:00:00 2001 From: Chris Neffshade Date: Tue, 12 Sep 2023 16:37:27 +0000 Subject: [PATCH] Fix arg. --- sdks/python/apache_beam/dataframe/frame_base.py | 2 +- sdks/python/apache_beam/dataframe/frames_test.py | 7 ------- 2 files changed, 1 insertion(+), 8 deletions(-) diff --git a/sdks/python/apache_beam/dataframe/frame_base.py b/sdks/python/apache_beam/dataframe/frame_base.py index ba06b925ac4c..88e6eb6a060b 100644 --- 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 index ae6439a94001..69e7f4b19d6f 100644 --- 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)