fix cell_block_id location in step4e

mikarubi · Dec 14, 2023 · 9e2b3af · 9e2b3af
1 parent c3eea3c
commit 9e2b3af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/voluseg/_steps/step4e.py b/voluseg/_steps/step4e.py
@@ -46,10 +46,10 @@ def add_data(tuple_ii):
             fullname_block = os.path.join(dir_cell, 'block%05d'%(ii))
             with h5py.File(fullname_block+hdf, 'r') as file_handle:
                 for ci in range(file_handle['n_cells'][()]):
-                    cell_block_id.append(ii)
                     cell_xyz.append(file_handle['/cell/%05d/xyz'%(ci)][()])
                     cell_weights.append(file_handle['/cell/%05d/weights'%(ci)][()])
                     cell_timeseries.append(file_handle['/cell/%05d/timeseries'%(ci)][()])
+                    cell_block_id.append(ii)
 
         except KeyError:
             print('block %d is empty.'%ii)