RF: Make sure that ASCII-encoded DataArrays are returned with expecte…

…d shape
simkarwin · Feb 23, 2024 · 6ffeeac · 6ffeeac
1 parent 79792de
commit 6ffeeac
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nibabel/gifti/parse_gifti_fast.py b/nibabel/gifti/parse_gifti_fast.py
@@ -74,6 +74,10 @@ def read_data_block(darray, fname, data, mmap):
         # GIFTI_ENCODING_ASCII
         c = StringIO(data)
         da = np.loadtxt(c, dtype=dtype)
+        # Reshape to dims specified in GiftiDataArray attributes, but preserve
+        # existing behaviour of loading as 1D for arrays with a dimension of
+        # length 1
+        da = da.reshape(darray.dims).squeeze()
         return da  # independent of the endianness
     elif enclabel not in ('B64BIN', 'B64GZ', 'External'):
         return 0