Merge branch 'fix-max-reads-num-reads' into 'master'

take min of max reads and num reads when max reads used See merge request machine-learning/bonito!171
nanoporetech · Apr 9, 2024 · 0ba190f · 0ba190f
2 parents 22f4ef8 + 073310c
commit 0ba190f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bonito/cli/basecaller.py b/bonito/cli/basecaller.py
@@ -120,6 +120,8 @@ def main(args):
 
     if args.max_reads:
         reads = take(reads, args.max_reads)
+        if num_reads is not None:
+            num_reads = min(num_reads, args.max_reads)
 
     if args.save_ctc:
         reads = (