Fix shuffle passing

zarr-developers · Jun 21, 2024 · 1abe5d5 · 1abe5d5
1 parent 16e1806
commit 1abe5d5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/numcodecs/blosc.py b/numcodecs/blosc.py
@@ -21,10 +21,10 @@ def cbuffer_complib(source):
     """Return the name of the compression library used to compress `source`."""
     return blosc2.get_clib(source)
 
-def compress(source, cname: bytes, clevel, shuffle=SHUFFLE, blocksize=AUTOBLOCKS):
+def compress(source, cname: bytes, clevel, shuffle: int=SHUFFLE, blocksize=AUTOBLOCKS):
     cname = cname.decode('ascii')
     blosc2.set_blocksize(blocksize)
-    return blosc2.compress(source, codec=getattr(blosc2.Codec, cname.upper()), clevel=clevel, filter=shuffle)
+    return blosc2.compress(source, codec=getattr(blosc2.Codec, cname.upper()), clevel=clevel, filter=_shuffles[shuffle])
 
 
 class Blosc(Codec):
@@ -66,7 +66,7 @@ def __init__(self, cname='lz4', clevel=5, shuffle=SHUFFLE, blocksize=AUTOBLOCKS)
 
     def encode(self, buf):
         buf = ensure_contiguous_ndarray(buf, self.max_buffer_size)
-        return compress(buf, bytes(self.cname, 'ascii'), self.clevel, _shuffles[self.shuffle], self.blocksize)
+        return compress(buf, bytes(self.cname, 'ascii'), self.clevel, self.shuffle, self.blocksize)
 
     def decode(self, buf, out=None):
         buf = ensure_contiguous_ndarray(buf, self.max_buffer_size)