fix(shard): enable floating support for create_sharded_image_info

seung-lab · Jan 29, 2025 · e5025d1 · e5025d1
1 parent 6b9d787
commit e5025d1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/igneous/task_creation/image.py b/igneous/task_creation/image.py
@@ -397,10 +397,14 @@ def create_sharded_image_info(
   """
   if isinstance(dtype, int):
     byte_width = dtype
-  elif isinstance(dtype, str) or np.issubdtype(dtype, np.integer):
+  elif (
+    isinstance(dtype, str) 
+    or np.issubdtype(dtype, np.integer) 
+    or np.issubdtype(dtype, np.floating)
+  ):
     byte_width = np.dtype(dtype).itemsize
   else:
-    raise ValueError(f"{dtype} must be int, str, or np.integer.")
+    raise ValueError(f"{dtype} must be int, str, np.integer, or np.floating.")
 
   voxels = prod(dataset_size)
   chunk_voxels = prod(chunk_size)