Skip to content

Commit 7ce7d0f

Browse files
committed
addressed matthew comment, and adding shrink to finalize_append step
1 parent 2ace3ef commit 7ce7d0f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

nibabel/streamlines/array_sequence.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def __init__(self, arr_seq, common_shape, dtype):
3737
self.common_shape = common_shape
3838
n_in_row = reduce(mul, common_shape, 1)
3939
bytes_per_row = n_in_row * dtype.itemsize
40-
self.rows_per_buf = int(np.ceil(self.bytes_per_buf / bytes_per_row))
40+
self.rows_per_buf = max(1, self.bytes_per_buf // bytes_per_row)
4141

4242
def update_seq(self, arr_seq):
4343
arr_seq._offsets = np.array(self.offsets)
@@ -185,6 +185,7 @@ def finalize_append(self):
185185
return
186186
self._build_cache.update_seq(self)
187187
self._build_cache = None
188+
self.shrink_data()
188189

189190
def _resize_data_to(self, n_rows, build_cache):
190191
""" Resize data array if required """

0 commit comments

Comments
 (0)