Rename pyarrow_dtype to pyarrow_type.

rapidsai · Mar 31, 2021 · 286b686 · 286b686
1 parent 040401c
commit 286b686
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/python/cudf/cudf/core/column/column.py b/python/cudf/cudf/core/column/column.py
@@ -1840,12 +1840,12 @@ def as_column(
         else:
             pyarrow_array = pa.array(arbitrary, from_pandas=nan_as_null)
             if isinstance(pyarrow_array.type, pa.Decimal128Type):
-                pyarrow_dtype = cudf.Decimal64Dtype.from_arrow(
+                pyarrow_type = cudf.Decimal64Dtype.from_arrow(
                     pyarrow_array.type
                 )
             else:
-                pyarrow_dtype = arbitrary.dtype
-            data = as_column(pyarrow_array, dtype=pyarrow_dtype)
+                pyarrow_type = arbitrary.dtype
+            data = as_column(pyarrow_array, dtype=pyarrow_type)
         if dtype is not None:
             data = data.astype(dtype)
 

diff --git a/python/cudf/cudf/tests/test_groupby.py b/python/cudf/cudf/tests/test_groupby.py
@@ -350,7 +350,8 @@ def test_groupby_2keys_agg(nelem, func):
 
 @pytest.mark.skipif(
     NVCC_MAJOR_VERSION < 11,
-    reason="These aggregations are not supported on CUDA 10.x.")
+    reason="These aggregations are not supported on CUDA 10.x.",
+)
 @pytest.mark.parametrize("num_groups", [2, 3, 10, 50, 100])
 @pytest.mark.parametrize("nelem_per_group", [1, 10, 100])
 @pytest.mark.parametrize(