Skip to content
Snippets Groups Projects
Commit d1a3ec49 authored by François  LAURENT's avatar François LAURENT
Browse files

fix for dict-like labels element

parent bad8ea67
No related branches found
No related tags found
1 merge request!2make_dataset.py extracts more time segments at the ends of tracks
......@@ -52,6 +52,8 @@ def make_dataset(backend, labels_expected=False, trxmat_only=False,
# note kwargs['labels'] may be defined, but we dismiss
# the input argument, because we need to preserve the
# order of the labels (the class indices)
if isinstance(labels, dict):
labels = labels['names']
kwargs['labels'] = labels
break
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment