potential bugfix for colored labels

src/maggotuba/models/modules.py

@@ -343,9 +343,12 @@ class MaggotClassifier(MaggotModule):
 
     @classmethod
     def load_model(cls, config, path):
+        labels = config["behavior_labels"]
+        if isinstance(labels, dict):
+            labels = labels['names']
         model = DeepLinear(
                 n_input=config["dim_latent"],
-                n_output=len(config["behavior_labels"]),
+                n_output=len(labels),
                 n_hidden=config["clf_depth"]*[None],
                 batch_norm=config["batch_norm"],
                 weight_init=config["weight_init"],