fix deadloop

usc-isi-i2 · Dec 7, 2018 · 086fde1 · 086fde1
1 parent eca04b4
commit 086fde1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/rltk/dataset.py b/rltk/dataset.py
@@ -190,6 +190,7 @@ def filter(self, condition: Callable, n: int = 10):
             if counter >= n:
                 break
             if condition(r):
+                counter += 1
                 yield r
 
     def __iter__(self):