Revert "Use df._data.columns."

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