Fix type hints (#7)

allen-cell-animated · Feb 15, 2023 · 001361e · 001361e
1 parent de185b0
commit 001361e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/abm_shape_collection/make_voxels_array.py b/src/abm_shape_collection/make_voxels_array.py
@@ -3,7 +3,7 @@
 
 
 @task
-def make_voxels_array(voxels: list[list], scale: int = 1) -> np.ndarray:
+def make_voxels_array(voxels: list[tuple[int, int, int]], scale: int = 1) -> np.ndarray:
     # Center voxels around (0,0,0).
     center_x, center_y, center_z = [round(x) for x in np.array(voxels).mean(axis=0)]
     all_xyz_centered = [(z - center_z, y - center_y, x - center_x) for x, y, z in voxels]