From bcc82edbba476786a2e61a70e06b3e8396ce2042 Mon Sep 17 00:00:00 2001 From: Hao-Ting Wang Date: Tue, 28 Jan 2025 13:39:25 -0500 Subject: [PATCH] FIX mypy type on mask array for sanity check (#189) --- giga_connectome/mask.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/giga_connectome/mask.py b/giga_connectome/mask.py index 63fd806..3418890 100644 --- a/giga_connectome/mask.py +++ b/giga_connectome/mask.py @@ -262,9 +262,11 @@ def _check_mask_affine( key_to_header[affine_hashable] = affine if isinstance(mask_imgs[0], Nifti1Image): - mask_arrays = np.arange(len(mask_imgs)) + # this looks janky, but this is for mypy to understand the type + mask_arrays = np.array(np.arange(len(mask_imgs)).tolist()) else: mask_arrays = np.array(mask_imgs) + # get most common values common_affine = max( set(header_info["affine"]), key=header_info["affine"].count