usecols is a list of strings

rapidsai · Nov 26, 2024 · 31d1212 · 31d1212
1 parent 187a43f
commit 31d1212
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/python/cudf_polars/cudf_polars/dsl/ir.py b/python/cudf_polars/cudf_polars/dsl/ir.py
@@ -493,10 +493,7 @@ def do_evaluate(
                 options.set_header(header)
                 options.set_dtypes(schema)
                 if usecols is not None:
-                    if all(isinstance(col, int) for col in usecols):
-                        options.set_use_cols_indexes(list(usecols))
-                    else:
-                        options.set_use_cols_names([str(name) for name in usecols])
+                    options.set_use_cols_names([str(name) for name in usecols])
                 options.set_na_values(null_values)
                 if comment is not None:
                     options.set_comment(comment)