Oups, label volume name

scil-vital · Feb 26, 2024 · 603f431 · 603f431
1 parent 6dd4a80
commit 603f431
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dwi_ml/data/dataset/streamline_containers.py b/dwi_ml/data/dataset/streamline_containers.py
@@ -53,9 +53,9 @@ def _load_connectivity_info(hdf_group: h5py.Group):
         contains_connectivity = True
         if 'connectivity_nb_blocs' in hdf_group.attrs:
             connectivity_nb_blocs = hdf_group.attrs['connectivity_nb_blocs']
-        elif 'connectivity_labels' in hdf_group:
+        elif 'connectivity_label_volume' in hdf_group:
             connectivity_labels = np.asarray(
-                hdf_group['connectivity_labels'], dtype=int)
+                hdf_group['connectivity_label_volume'], dtype=int)
         else:
             raise ValueError(
                 "Information stored in the hdf5 is that it contains a "