Update input_mod.py

theosanderson · Jan 21, 2023 · 0ce2a8e · 0ce2a8e
1 parent c0d2531
commit 0ce2a8e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/chronumental/input_mod.py b/src/chronumental/input_mod.py
@@ -19,9 +19,9 @@ def read_tabular_file(tabular_file_name, **kwargs):
     stripped_name = tabular_file_name.replace(".gz", "").replace(
         ".bz2", "").replace(".xz", "").replace(".lzma", "")
     if stripped_name.endswith(".csv"):
-        return pd.read_csv(tabular_file, **kwargs)
+        return pd.read_csv(tabular_file, dtype={"strain": str, "name": str, "taxon": str}, **kwargs)
     if stripped_name.endswith(".tsv"):
-        return pd.read_csv(tabular_file, sep="\t", **kwargs)
+        return pd.read_csv(tabular_file, sep="\t", dtype={"strain": str, "name": str, "taxon": str},**kwargs)
     raise Exception(
         f"Tabular file {tabular_file_name} was expected to end in tsv or csv")