diff --git a/nd2reader/model/image.py b/nd2reader/model/image.py index 4bbc0ad..594b0b0 100644 --- a/nd2reader/model/image.py +++ b/nd2reader/model/image.py @@ -19,6 +19,12 @@ class Image(np.ndarray): self._channel = None self._z_level = None + def __array_wrap__(self, obj): + if len(obj.shape) == 0: + return obj[()] + else: + return np.ndarray.__array_wrap__(obj) + def add_params(self, index, timestamp, frame_number, field_of_view, channel, z_level): """ :param index: The integer that can be used to directly index this image