fix error

Robaina · Nov 3, 2023 · 2573960 · 2573960
1 parent 0a3d4d8
commit 2573960
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pynteny/hmm.py b/src/pynteny/hmm.py
@@ -146,6 +146,8 @@ def __init__(
         self._meta = pd.read_csv(
             self._metadata_file, sep="\t", usecols=metadata_columns
         )
+        if "#ncbi_accession" in self._meta.columns:
+            self._meta = self._meta.rename(columns={"#ncbi_accession": "accession"})
 
     @property
     def meta(self) -> pd.DataFrame:
@@ -238,8 +240,6 @@ class PGAP(HMMDatabase):
     def __init__(self, *args, **kwargs):
         """Initialize class PGAP"""
         super().__init__(*args, **kwargs)
-        self._meta = self._meta.rename(columns={"#ncbi_accession": "accession"})
-        # self._meta = self.remove_missing_HMMs_from_metadata(meta_outfile=None)
 
     def remove_missing_HMMs_from_metadata(
         self, hmm_database: Path, meta_outfile: Path = None