diff --git a/PopPUNK/prune_db.py b/PopPUNK/prune_db.py index 51a921b4..70f0fa8a 100755 --- 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 index cd2411d9..68b00383 100755 --- 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: