fix(datasets): add save_dataframe method

entelecheia · Jun 23, 2023 · 9633738 · 9633738
1 parent ee4a166
commit 9633738
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/src/hyfi/main/__init__.py b/src/hyfi/main/__init__.py
@@ -718,6 +718,31 @@ def load_dataframe(
             data_file, data_dir, filetype, columns, index_col, verbose, **kwargs
         )
 
+    @staticmethod
+    def save_dataframe(
+        data: pd.DataFrame,
+        data_file: str,
+        data_dir: str = "",
+        columns: Optional[Sequence[str]] = None,
+        index: bool = False,
+        filetype: str = "parquet",
+        suffix: str = "",
+        verbose: bool = False,
+        **kwargs,
+    ):
+        """Save data to a file"""
+        Datasets.save_dataframes(
+            data,
+            data_file,
+            data_dir,
+            columns,
+            index,
+            filetype,
+            suffix,
+            verbose,
+            **kwargs,
+        )
+
     @staticmethod
     def save_dataframes(
         data: Union[pd.DataFrame, dict],