Fixing bugs

Author: thodson-usgs

numcodecs/__init__.py

@@ -91,6 +91,9 @@
 from numcodecs.shuffle import Shuffle
 register_codec(Shuffle)
 
+from numcodecs.bitinfo import BitInfo
+register_codec(BitInfo)
+
 from numcodecs.bitround import BitRound
 register_codec(BitRound)
 

numcodecs/bitinfo.py

@@ -50,6 +50,8 @@ def encode(self, buf):
         compressible.
         """
         a = ensure_ndarray_like(buf)
+        dtype = a.dtype
+
         if not a.dtype.kind == "f" or a.dtype.itemsize > 8:
             raise TypeError("Only float arrays (16-64bit) can be bit-rounded")
 
@@ -70,7 +72,7 @@ def encode(self, buf):
 
         keepbits = max(keepbits)
 
-        return BitRound._bitround(a, keepbits)
+        return BitRound._bitround(a, keepbits, dtype)
 
 
 def exponent_bias(dtype):

numcodecs/bitround.py

@@ -59,11 +59,11 @@ def encode(self, buf):
         if self.keepbits > bits:
             raise ValueError("Keepbits too large for given dtype")
 
-        return self._bitround(a, self.keepbits)
+        return self._bitround(a, self.keepbits, a.dtype)
 
     @staticmethod
-    def _bitround(buf, keepbits):
-        bits = max_bits[str(buf.dtype)]
+    def _bitround(buf, keepbits, dtype):
+        bits = max_bits[str(dtype)]
         a_int_dtype = np.dtype(buf.dtype.str.replace("f", "i"))
         all_set = np.array(-1, dtype=a_int_dtype)
         b = buf.view(a_int_dtype)