rename n_reads->max-reads

nanoporetech · Jan 20, 2021 · a11be1c · a11be1c
1 parent 494f495
commit a11be1c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bonito/cli/basecaller.py b/bonito/cli/basecaller.py
@@ -8,6 +8,7 @@
 from tqdm import tqdm
 from time import perf_counter
 from datetime import timedelta
+from itertools import islice as take
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
 
 from bonito.aligner import Aligner
@@ -38,9 +39,8 @@ def main(args):
         args.reads_directory, n_proc=8, recursive=args.recursive,
         read_ids=column_to_set(args.read_ids), skip=args.skip,
     )
-    if args.n_reads:
-        from itertools import islice as take
-        reads = take(reads, args.n_reads)
+    if args.max_reads:
+        reads = take(reads, args.max_reads)
 
     basecall = load_symbol(args.model_directory, "basecall")
 
@@ -89,5 +89,5 @@ def argparser():
     parser.add_argument("--ctc-min-coverage", default=0.9, type=float)
     parser.add_argument("--ctc-min-accuracy", default=0.9, type=float)
     parser.add_argument("--chunksize", default=4000, type=int)
-    parser.add_argument("--n_reads", default=0, type=int)
+    parser.add_argument("--max-reads", default=0, type=int)
     return parser