Skip to content

Commit f3ec45a

Browse files
committed
fix: Add writelines proxy method
1 parent f0d32a0 commit f3ec45a

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

nibabel/openers.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
if ty.TYPE_CHECKING:
2222
from types import TracebackType
2323

24-
from _typeshed import WriteableBuffer
24+
from _typeshed import ReadableBuffer, WriteableBuffer
2525

2626
from ._typing import Self
2727

@@ -231,6 +231,9 @@ def readinto(self, buffer: WriteableBuffer, /) -> int | None:
231231
def write(self, b: bytes, /) -> int | None:
232232
return self.fobj.write(b)
233233

234+
def writelines(self, lines: ty.Iterable[ReadableBuffer], /) -> None:
235+
self.fobj.writelines(lines)
236+
234237
def seek(self, pos: int, whence: int = 0, /) -> int:
235238
return self.fobj.seek(pos, whence)
236239

nibabel/volumeutils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1364,7 +1364,7 @@ def shape_zoom_affine(
13641364
return aff
13651365

13661366

1367-
def rec2dict(rec: np.ndarray) -> dict[str, np.generic | np.ndarray]:
1367+
def rec2dict(rec: np.record) -> dict[str, np.generic | np.ndarray]:
13681368
"""Convert recarray to dictionary
13691369
13701370
Also converts scalar values to scalars
@@ -1387,7 +1387,7 @@ def rec2dict(rec: np.ndarray) -> dict[str, np.generic | np.ndarray]:
13871387
True
13881388
"""
13891389
dct = {}
1390-
for key in rec.dtype.fields:
1390+
for key in rec.dtype.fields or ():
13911391
val = rec[key]
13921392
try:
13931393
val = val.item()

0 commit comments

Comments
 (0)