pass kwargs in cudf_read_csv

mars-project · Dec 12, 2019 · a0372be · a0372be
1 parent 61d9bfd
commit a0372be
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mars/dataframe/datasource/read_csv.py b/mars/dataframe/datasource/read_csv.py
@@ -212,8 +212,9 @@ def _pandas_read_csv(cls, f, op):
 
     @classmethod
     def _cudf_read_csv(cls, op):
+        csv_kwargs = op.extra_params
         df = cudf.read_csv(op.path, byte_range=(op.offset, op.size), sep=op.sep, names=op.names,
-                           dtype=cls._validate_dtypes(op.outputs[0].dtypes, op.gpu))
+                           dtype=cls._validate_dtypes(op.outputs[0].dtypes, op.gpu), **csv_kwargs)
         return df
 
     @classmethod