Update src/nifreeze/utils/iterators.py

Co-authored-by: Oscar Esteban <[email protected]>

Author: jhlegarreta

src/nifreeze/utils/iterators.py

@@ -26,7 +26,7 @@
 from itertools import chain, zip_longest
 from typing import Iterator
 
-SIZE_KEYS = ("bvals", "uptake")
+SIZE_KEYS = ("size", "bvals", "uptake")
 """Keys that may be used to infer the number of volumes in a dataset. These keys
 correspond to properties that distinguish one imaging modality from another, and
 are part of the 4th axis (e.g. diffusion gradients in DWI or update in PET)."""