Merge pull request #21 from tcchiao/patch-1

Joe Hamman · web-flow · commit 67d847afa08a · 2021-06-14T15:51:25.000-07:00
allow equal stride and dimsize
diff --git a/xbatcher/generators.py b/xbatcher/generators.py
@@ -19,7 +19,7 @@ def _slices(dimsize, size, overlap=0):
     slices = []
     stride = size - overlap
     assert stride > 0
-    assert stride < dimsize
+    assert stride <= dimsize
     for start in range(0, dimsize, stride):
         end = start + size
         if end <= dimsize:
