Fix dataframe.dtype AttributeError

rapidsai · Apr 12, 2022 · 1509af5 · 1509af5
1 parent 84f096c
commit 1509af5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/python/cudf/cudf/core/dataframe.py b/python/cudf/cudf/core/dataframe.py
@@ -5146,9 +5146,7 @@ def _reduce(
                     for col in source._data.names
                 ]
             except AttributeError:
-                raise TypeError(
-                    f"cannot perform {op} with type {source.dtype}"
-                )
+                raise TypeError(f"Not all column dtypes support op {op}")
 
             return Series._from_data(
                 {None: result}, as_index(source._data.names)

diff --git a/python/cudf/cudf/tests/test_dataframe.py b/python/cudf/cudf/tests/test_dataframe.py
@@ -9172,7 +9172,7 @@ def test_mean_timeseries():
 
     assert_eq(expected, actual)
 
-    with pytest.raises(AttributeError):
+    with pytest.raises(TypeError):
         gdf.mean()
 
 
@@ -9195,5 +9195,5 @@ def test_std_different_dtypes(data):
 
     assert_eq(expected, actual)
 
-    with pytest.raises(AttributeError):
+    with pytest.raises(TypeError):
         gdf.std()