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 %{