Connectivity labels volume as group's dataset

scil-vital · Feb 22, 2024 · 5baadca · 5baadca
1 parent 2cb9562
commit 5baadca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dwi_ml/data/hdf5/hdf5_creation.py b/dwi_ml/data/hdf5/hdf5_creation.py
@@ -593,8 +593,8 @@ def _create_streamline_groups(self, ref, subj_input_dir, subj_id,
                 streamlines_group.create_dataset(
                     'connectivity_matrix', data=connectivity_matrix)
                 if conn_info[0] == 'from_label':
-                    streamlines_group.attrs['connectivity_label_volume'] = \
-                        conn_info[1]
+                    streamlines_group.create_dataset(
+                        'connectivity_label_volume', data=conn_info[1])
                 else:
                     streamlines_group.attrs['connectivity_nb_blocs'] = \
                         conn_info[1]