Fix bad header check

RasmussenLab · Dec 3, 2024 · 80bd1e6 · 80bd1e6
1 parent 104dfba
commit 80bd1e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vamb/__main__.py b/vamb/__main__.py
@@ -396,7 +396,7 @@ def get_taxonomy(args: argparse.Namespace) -> Union[RefinedTaxonomy, UnrefinedTa
         )
     with open(check_existing_file(path)) as file:
         try:
-            header = next(file)
+            header = next(file).rstrip("\r\n")
         except StopIteration:
             header = None
 
@@ -410,7 +410,7 @@ def get_taxonomy(args: argparse.Namespace) -> Union[RefinedTaxonomy, UnrefinedTa
         raise ValueError(
             f'ERROR: When reading taxonomy file at "{path}", '
             f"the first line was not either {repr(vamb.taxonomy.TAXONOMY_HEADER)} "
-            f"or {repr(vamb.taxonomy.TAXONOMY_HEADER)}'"
+            f"or {repr(vamb.taxonomy.PREDICTED_TAXONOMY_HEADER)}'"
         )