Add and clean

yuanx749 · May 28, 2024 · 0f939af · 0f939af
1 parent 7a3da73
commit 0f939af
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/pycdhit/__init__.py b/pycdhit/__init__.py
@@ -4,7 +4,7 @@
 from ._commands import *  # noqa: F403
 from ._io import *  # noqa: F403
 
-VERSION = "0.12.0"
+VERSION = "0.13.0"
 
 __all__ = [  # noqa: F405
     "CommandBase",

diff --git a/pycdhit/_class.py b/pycdhit/_class.py
@@ -80,11 +80,17 @@ class CDHIT(CommandBase):
 
     """
 
+    _progs = {
+        "cd-hit",
+        "cd-hit-2d",
+        "cd-hit-est",
+        "cd-hit-est-2d",
+        "cd-hit-454",
+    }
+
     def __init__(self, prog: str = "cd-hit", path: str = None):
-        if prog not in {"cd-hit", "cd-hit-2d", "cd-hit-est", "cd-hit-est-2d"}:
-            raise ValueError(
-                f"'{prog}' not in {{'cd-hit', 'cd-hit-2d', 'cd-hit-est', 'cd-hit-est-2d'}}"  # noqa: E501
-            )
+        if prog not in self._progs:
+            raise ValueError(f"'{prog}' not in {self._progs}")
         super().__init__(prog, path)
 
     def help(self):