cleanup

rapidsai · Jan 19, 2022 · a65d4be · a65d4be
1 parent 3c1447d
commit a65d4be
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/python/cudf/cudf/_fuzz_testing/tests/fuzz_test_orc.py b/python/cudf/cudf/_fuzz_testing/tests/fuzz_test_orc.py
@@ -18,7 +18,7 @@
     data_handle=OrcReader,
     params={
         "columns": ALL_POSSIBLE_VALUES,
-        "skiprows": [None],
+        "skiprows": ALL_POSSIBLE_VALUES,
         "num_rows": ALL_POSSIBLE_VALUES,
         "use_index": ALL_POSSIBLE_VALUES,
     },

diff --git a/python/cudf/cudf/testing/dataset_generator.py b/python/cudf/cudf/testing/dataset_generator.py
@@ -805,15 +805,7 @@ def struct_generator(dtype, cardinality, size, max_null_frequency):
 
 
 def create_nested_struct_type(max_types_at_each_level, nesting_level):
-    dtypes_list = cudf.utils.dtypes.ALL_TYPES - {
-        "category",
-        "datetime64[ns]",
-        "str",
-    } - cudf.utils.dtypes.TIMEDELTA_TYPES - {
-        "uint32"
-    } - cudf.utils.dtypes.UNSIGNED_TYPES | {
-        "struct"
-    }
+    dtypes_list = cudf.utils.dtypes.ALL_TYPES
     picked_types = np.random.choice(list(dtypes_list), max_types_at_each_level)
     type_dict = {}
     for name, type_ in enumerate(picked_types):