Bug fix: correctly catch loos::StopIteration exceptions for AtomicGro…

…upIterator
GrossfieldLab · Oct 24, 2022 · 015bba6 · 015bba6
1 parent 852c866
commit 015bba6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/AtomicGroup.i b/src/AtomicGroup.i
@@ -71,7 +71,7 @@
 
  pAtom __next__() {
  if (_idx >= _ag->size())
- throw(loos::StopIteration());
+ throw(StopIteration());
  return((*_ag)[_idx++]);
  }
 
@@ -85,6 +85,7 @@
 
 %}
 
+%catches(loos::StopIteration) loos::AtomicGroupPythonIterator::__next__();
 
 // Translate C++ exception into Python's
 %typemap(throws) loos::StopIteration %{