small bugfix

uio-bmi · Nov 6, 2023 · 4a51a66 · 4a51a66
1 parent 0101f2d
commit 4a51a66
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/immuneML/simulation/dataset_generation/RandomDatasetGenerator.py b/immuneML/simulation/dataset_generation/RandomDatasetGenerator.py
@@ -167,7 +167,8 @@ def generate_receptor_dataset(receptor_count: int, chain_1_length_probabilities:
                                          for label, label_dict in labels.items()}, **{"subject": f"subj_{i + 1}"}})
                      for i in range(receptor_count)]
 
-        return ReceptorDataset.build_from_objects(receptors, 100, path, labels=labels)
+        return ReceptorDataset.build_from_objects(receptors, 100, path, labels={label: list(label_dict.keys())
+                                                                                for label, label_dict in labels.items()})
 
     @staticmethod
     def _check_sequence_dataset_generation_params(receptor_count: int, length_probabilities: dict, labels: dict,