minor bug fix to ensure slicing with integers

Chilipp · Jan 28, 2019 · 75a148e · 75a148e
1 parent 0822c13
commit 75a148e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/straditize/binary.py b/straditize/binary.py
@@ -1375,10 +1375,10 @@ def _filter_lines(self, locs, min_lw=1, max_lw=None):
         except ValueError:  # probably none of the lines is thick enough
             if all(len(indices) < min_lw for indices in np.split(
                     locs, np.where(np.diff(locs) != 1)[0]+1)):
-                return np.array([])
+                selection = np.array([], dtype=int)
             else:
                 raise
-        return selection
+        return np.asarray(selection, dtype=int)
 
     def recognize_xaxes(self, fraction=0.3, min_lw=1, max_lw=None,
                         remove=False, **kwargs):