add read_csv function

KoslickiLab · Oct 2, 2023 · 2614f16 · 2614f16
1 parent 99a240c
commit 2614f16
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/fununifrac/reproducibility/plot_mds.py b/fununifrac/reproducibility/plot_mds.py
@@ -23,8 +23,10 @@ def parsearg():
 
 if __name__ == "__main__":
     args = parsearg()
-
-    metadata = pd.read_table(args.metadata_file)
+    if args.metadata_file.endswith('.csv'):
+        metadata = pd.read_csv(args.metadata_file)
+    else:
+        metadata = pd.read_table(args.metadata_file)
     with open(args.label_file, 'r') as f:
         labels = f.readlines()
         labels = [l.strip() for l in labels]