Fix sketchlib functions in prune_db and reference_pick

bacpop · Feb 10, 2020 · 0bb9724 · 0bb9724
1 parent 620aa4a
commit 0bb9724
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/PopPUNK/prune_db.py b/PopPUNK/prune_db.py
@@ -12,6 +12,7 @@
 # import poppunk package
 from .__init__ import __version__
 
+from .sketchlib import no_sketchlib
 from .sketchlib import removeFromDB
 
 from .mash import checkMashVersion
@@ -135,6 +136,9 @@ def main():
     args = get_options()
 
     resketch = args.resketch
+    if no_sketchlib:
+        args.use_mash = True
+
     if args.use_mash:
         checkMashVersion(args.mash)
     else:

diff --git a/PopPUNK/reference_pick.py b/PopPUNK/reference_pick.py
@@ -12,6 +12,9 @@
 # import poppunk package
 from .__init__ import __version__
 
+from .sketchlib import no_sketchlib
+from .sketchlib import removeFromDB
+
 from .mash import checkMashVersion
 from .mash import createDatabaseDir
 from .mash import constructDatabase
@@ -63,7 +66,10 @@ def main():
 
     # Check input args ok
     args = get_options()
-    resketch = !(args.no_resketch)
+    resketch = not args.no_resketch
+    if no_sketchlib:
+        args.use_mash = True
+
     if args.use_mash:
         checkMashVersion(args.mash)
     else: