From d284eea77e13ccd31b324b08bdca70823d23ba78 Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Wed, 11 Nov 2020 07:43:30 -0800 Subject: [PATCH] style --- python/cudf/cudf/_fuzz_testing/utils.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/python/cudf/cudf/_fuzz_testing/utils.py b/python/cudf/cudf/_fuzz_testing/utils.py index 92e8082193b..2e929b0a148 100644 --- a/python/cudf/cudf/_fuzz_testing/utils.py +++ b/python/cudf/cudf/_fuzz_testing/utils.py @@ -173,7 +173,9 @@ def get_avro_schema(df): def get_orc_schema(df): - ordered_dict = OrderedDict((col_name, col_dtype) for col_name, col_dtype in df.dtypes.items()) + ordered_dict = OrderedDict( + (col_name, col_dtype) for col_name, col_dtype in df.dtypes.items() + ) schema = pyorc.Struct(**ordered_dict) return schema @@ -236,7 +238,10 @@ def _null_to_None(value): ] ) - tuple_list = [tuple(map(_null_to_None, tup)) if has_nulls_or_nullable_dtype else tup for tup in df.itertuples(index=False, name=None)] + tuple_list = [ + tuple(map(_null_to_None, tup)) if has_nulls_or_nullable_dtype else tup + for tup in df.itertuples(index=False, name=None) + ] return tuple_list @@ -272,7 +277,9 @@ def orc_to_pandas(file_name=None, file_io_obj=None, stripes=None): reader, columns=reader.schema.fields.keys() ) else: - records = [record for i in stripes for record in list(reader.read_stripe(i))] + records = [ + record for i in stripes for record in list(reader.read_stripe(i)) + ] df = pd.DataFrame.from_records( records, columns=reader.schema.fields.keys() )