csv backward compatibility

shayne-longpre · Nov 19, 2020 · 9edd2d4 · 9edd2d4
1 parent 7478935
commit 9edd2d4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/datasets/csv/csv.py b/datasets/csv/csv.py
@@ -21,6 +21,7 @@ class CsvConfig(datasets.BuilderConfig):
     delimiter: Optional[str] = None
     header: Optional[Union[int, List[int], str]] = "infer"
     names: Optional[List[str]] = None
+    column_names: Optional[List[str]] = None
     index_col: Optional[Union[int, str, List[int], List[str]]] = None
     usecols: Optional[Union[List[int], List[str]]] = None
     prefix: Optional[str] = None
@@ -57,6 +58,8 @@ class CsvConfig(datasets.BuilderConfig):
     def __post_init__(self):
         if self.delimiter is not None:
             self.sep = self.delimiter
+        if self.column_names is not None:
+            self.names = self.column_names
 
 
 class Csv(datasets.ArrowBasedBuilder):