feat(gunc): Updated GUNC paths.

aaronmussig · Mar 17, 2022 · 8306b31 · 8306b31
1 parent 5d0e84f
commit 8306b31
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/magna/gunc/__init__.py b/magna/gunc/__init__.py
@@ -10,23 +10,23 @@
 
 def gunc_max_css_scores_gtdb_r95() -> pd.DataFrame:
     """Return the max clade separation score (CSS) for the R95 GTDB."""
-    path = os.path.join(MAGNA_DIR, 'dataset', 'gunc', 'GUNC.gtdb_95.maxCSS_level.feather')
+    path = os.path.join(MAGNA_DIR, 'gunc', 'gtdb_95.maxcss_level.feather')
     if not os.path.isfile(path):
         raise IOError(f'{path} does not exist.')
     return pd.read_feather(path)
 
 
 def gunc_contig_assignment_gtdb_r95() -> pd.DataFrame:
     """Return the contig assignment for the R95 GTDB."""
-    path = os.path.join(MAGNA_DIR, 'dataset', 'gunc', 'GUNC.gtdb_95.contig_assignments.feather')
+    path = os.path.join(MAGNA_DIR,'gunc', 'GUNC.gtdb_95.contig_assignments.feather')
     if not os.path.isfile(path):
         raise IOError(f'{path} does not exist.')
     return pd.read_feather(path)
 
 
 def gunc_all_levels_gtdb_r95() -> pd.DataFrame:
     """Return GUNC output at all levels for the R95 GTDB."""
-    path = os.path.join(MAGNA_DIR, 'dataset', 'gunc', 'gtdb_95.all_levels.feather')
+    path = os.path.join(MAGNA_DIR, 'gunc', 'gtdb_95.all_levels.feather')
     if not os.path.isfile(path):
         raise IOError(f'{path} does not exist.')
     return pd.read_feather(path)