From 0c776eba61937dc9f726a47ba5a46dc700b4474f Mon Sep 17 00:00:00 2001 From: Daniel Weindl Date: Mon, 16 Jan 2023 15:25:38 +0100 Subject: [PATCH] Make `ReturnDataView.status` return `int` (#1929) --- python/sdist/amici/numpy.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/sdist/amici/numpy.py b/python/sdist/amici/numpy.py index 65df16a557..9b61bfaea4 100644 --- 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,