allow to set has_header

cohere/custom_model_dataset.py

@@ -87,18 +87,20 @@ class CsvDataset(LocalFileCustomModelDataset):
     another prompt, another completion
     """
 
-    def __init__(self, train_file: str, delimiter: str, eval_file: Optional[str] = None):
+    def __init__(self, train_file: str, delimiter: str, eval_file: Optional[str] = None, has_header: Optional[bool] = False):
         """
         Args:
             train_file (str): local path to csv with training data
             eval_file: (str, optional): local path to a csv with eval data
         """
         super().__init__(train_file, eval_file)
         self._delimiter = delimiter
+        self._has_header = has_header
 
     def file_config(self) -> FileConfig:
         config = dict(_empty_file_config)
         config["delimiter"] = self._delimiter
+        config["hasHeader"] = self._has_header
         return config