From cd641e30d15f4b2596a36f716a97866f88cd1a9f Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Fri, 29 Apr 2022 08:25:06 -0500 Subject: [PATCH] Use df._data.columns. --- python/cudf/cudf/_fuzz_testing/utils.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/python/cudf/cudf/_fuzz_testing/utils.py b/python/cudf/cudf/_fuzz_testing/utils.py index 9f3c0ab6d5f..b5fdbbf3203 100644 --- 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: