Make ReturnDataView.status return int (#1929)

AMICI-dev · Jan 16, 2023 · 0c776eb · 0c776eb
1 parent 6e44a70
commit 0c776eb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/sdist/amici/numpy.py b/python/sdist/amici/numpy.py
@@ -235,9 +235,13 @@ def __getitem__(self, item: str) -> Union[np.ndarray, ReturnDataPtr,
 
         :returns: self[item]
         """
+        if item == 'status':
+            return int(super().__getitem__(item))
+
         if item == 't':
             item = 'ts'
-        return super(ReturnDataView, self).__getitem__(item)
+
+        return super().__getitem__(item)
 
     def by_id(
             self,