Fix exception handling in __next__

LiberTEM · Dec 6, 2023 · 35ca25b · 35ca25b
1 parent c51ea2d
commit 35ca25b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/libertem_live/api.py b/src/libertem_live/api.py
@@ -39,10 +39,9 @@ def __iter__(self):
     def __next__(self):
         try:
             return next(self._inner_iter)
-        except StopIteration:
-            raise
-        finally:
+        except Exception:
             self._callback()
+            raise
 
     def close(self):
         try: