handle stitching single chunks shorter than the chunksize

nanoporetech · May 15, 2022 · 4585b74 · 4585b74
1 parent a49f924
commit 4585b74
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bonito/crf/basecall.py b/bonito/crf/basecall.py
@@ -19,6 +19,8 @@ def stitch_results(results, length, size, overlap, stride, reverse=False):
             k: stitch_results(v, length, size, overlap, stride, reverse=reverse)
             for k, v in results.items()
         }
+    if length < size:
+        return results[0, :int(np.floor(length / stride))]
     return stitch(results, size, overlap, length, stride, reverse=reverse)