Support TSV files in file_io (#341)

* done * done * support save tsv --------- Co-authored-by: Sagi Polaczek <[email protected]>
BiomedSciAI · Jan 28, 2024 · a58d869 · a58d869
1 parent 995d85a
commit a58d869
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/fuse/utils/file_io/file_io.py b/fuse/utils/file_io/file_io.py
@@ -291,6 +291,7 @@ def save_dataframe(df: pd.DataFrame, filename: str, **kwargs: dict) -> None:
 
  assert file_type in [
  "csv",
+ "tsv",
  "hd5",
  "hdf5",
  "hdf",
@@ -304,6 +305,8 @@ def save_dataframe(df: pd.DataFrame, filename: str, **kwargs: dict) -> None:
  df.to_pickle(filename, **kwargs)
  elif file_type == "csv":
  df.to_csv(filename, **kwargs)
+ elif file_type == "tsv":
+ df.to_csv(filename, sep="\t" ** kwargs)
  elif file_type in ["hd5", "hdf5", "hdf"]:
  df.to_hdf(filename, **kwargs)
  elif file_type == "xslx":
@@ -322,6 +325,7 @@ def read_dataframe(filename: str) -> pd.DataFrame:
 
  assert file_type in [
  "csv",
+ "tsv",
  "hd5",
  "hdf5",
  "hdf",
@@ -334,6 +338,8 @@ def read_dataframe(filename: str) -> pd.DataFrame:
  df = pd.read_pickle(filename)
  elif file_type == "csv":
  df = pd.read_csv(filename)
+ elif file_type == "tsv":
+ df = pd.read_csv(filename, sep="\t")
  elif file_type in ["hd5", "hdf5", "hdf"]:
  df = pd.read_hdf(filename)
  elif file_type == "xslx":