Use df._data.columns.

bdice · Apr 29, 2022 · cd641e3 · cd641e3
1 parent 1f037b0
commit cd641e3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/cudf/cudf/_fuzz_testing/utils.py b/python/cudf/cudf/_fuzz_testing/utils.py
@@ -312,9 +312,8 @@ def sanitize(value, struct_type):
         return tuple(values_list)
 
     has_nulls_or_nullable_dtype = any(
-        (col := df[colname]).dtype in pandas_dtypes_to_np_dtypes
-        or col.isnull().any()
-        for colname in df.columns
+        col.dtype in pandas_dtypes_to_np_dtypes or col.isnull().any()
+        for col in df._data.columns
     )
     pdf = df.copy(deep=True)
     for field in arrow_table_schema: