Skip to content

Commit 5f37398

Browse files
authored
Merge pull request #1253 from blakedewey/fix_spatial_image_typing
TYP: Annotate SpatialImage accepting affine=None
2 parents 1843355 + 21113e3 commit 5f37398

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

nibabel/spatialimages.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -481,7 +481,7 @@ class SpatialImage(DataobjImage):
481481
def __init__(
482482
self,
483483
dataobj: ArrayLike,
484-
affine: np.ndarray,
484+
affine: np.ndarray | None,
485485
header: FileBasedHeader | ty.Mapping | None = None,
486486
extra: ty.Mapping | None = None,
487487
file_map: FileMap | None = None,
@@ -566,6 +566,7 @@ def update_header(self) -> None:
566566

567567
def _affine2header(self) -> None:
568568
"""Unconditionally set affine into the header"""
569+
assert self._affine is not None
569570
RZS = self._affine[:3, :3]
570571
vox = np.sqrt(np.sum(RZS * RZS, axis=0))
571572
hdr = self._header

0 commit comments

Comments
 (0)