From 1a7d06a3be517f524071e5f8d61714d62baa4ceb Mon Sep 17 00:00:00 2001 From: Dimitri Yatsenko Date: Wed, 28 Feb 2024 11:06:26 +0000 Subject: [PATCH] minor bugfix: hard-coded encoded_dtype --- src/poisson_numcodecs/Poisson.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/poisson_numcodecs/Poisson.py b/src/poisson_numcodecs/Poisson.py index da0a385..65f2be4 100644 --- a/src/poisson_numcodecs/Poisson.py +++ b/src/poisson_numcodecs/Poisson.py @@ -9,7 +9,7 @@ ### NUMCODECS Codec ### class Poisson(Codec): - """Codec for 3-dimensional Delta. The codec assumes that input data are of shape: + """Codec for 3-dimensional data. The codec assumes that input data are of shape: (time, x, y). Parameters @@ -46,7 +46,7 @@ def decode(self, buf: bytes, out=None) -> np.array: inverse = estimate.make_inverse_lookup(lookup) ndims = int(buf[0]) shape = [int(_) for _ in buf[1:ndims+1]] - arr = np.frombuffer(buf[ndims+1:], dtype='uint8').reshape(shape) + arr = np.frombuffer(buf[ndims+1:], dtype=self.encoded_dtype).reshape(shape) decoded = estimate.lookup(arr, inverse) return decoded.astype(self.decoded_dtype)