allow virus arg to be passed as int (e.g. taxid)

pachterlab · Oct 17, 2024 · b81030f · b81030f
1 parent 6f82b94
commit b81030f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gget/gget_ncbi_virus.py b/gget/gget_ncbi_virus.py
@@ -642,6 +642,8 @@ def ncbi_virus(
         raise TypeError(
             "Argument 'proteins_complete' must be a boolean (True or False) or None."
         )
+    if isinstance(virus, int):
+        virus = str(virus)
 
     check_min_max(
         min_seq_length,