From d1a3ec49f83513915918bc1f0ed95daa6ab995a0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Fri, 30 Jun 2023 20:09:55 +0200
Subject: [PATCH] fix for dict-like labels element
---
src/maggotuba/data/make_dataset.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/maggotuba/data/make_dataset.py b/src/maggotuba/data/make_dataset.py
index 4687ff7..8981236 100644
--- a/src/maggotuba/data/make_dataset.py
+++ b/src/maggotuba/data/make_dataset.py
@@ -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
--
GitLab