take min of max reads and num reads when max reads used

nanoporetech · Mar 31, 2024 · 073310c · 073310c
1 parent b571854
commit 073310c
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 = (