[BUGFIX] Fix labeled components if initialized

2021-06-11 16:09:51 +02:00
parent 47d7f5831f
commit aaad2b8626
1 changed files with 1 additions and 1 deletions
--- a/prototorch/components/components.py
+++ b/prototorch/components/components.py
@@ -118,7 +118,7 @@ class LabeledComponents(Components):
            components, component_labels = parse_data_arg(
                initialized_components)
            super().__init__(initialized_components=components)
-            self._labels = component_labels
+            self._register_labels(component_labels)
        else:
            labels = get_labels_object(distribution)
            self.initial_distribution = labels.distribution