..

dweindl · Apr 16, 2024 · 1c0d719 · 1c0d719
1 parent b1de5cd
commit 1c0d719
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/swig/edata.i b/swig/edata.i
@@ -20,7 +20,9 @@ using namespace amici;
         from amici.numpy import ReturnDataView
 
         # Get the raw pointer if necessary
-        if isinstance(args[0], (ExpData, ExpDataPtr, ReturnDataView, Model, ModelPtr)):
+        if isinstance(args[0], (ExpData, ExpDataPtr, Model, ModelPtr)):
+            args = (_get_ptr(args[0]), *args[1:])
+        elif isinstance(args[0], ReturnDataView):
             args = (_get_ptr(args[0]["ptr"]), *args[1:])
 %}