From 0ce2a8e799229b4785e81c20bfd53802941cad09 Mon Sep 17 00:00:00 2001 From: Theo Sanderson Date: Sat, 21 Jan 2023 19:08:42 +0000 Subject: [PATCH] Update input_mod.py --- src/chronumental/input_mod.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/chronumental/input_mod.py b/src/chronumental/input_mod.py index 74b1567..b51c358 100644 --- 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")