julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[BUGFIX] Fix labeled components if initialized

Browse Source
This commit is contained in:
Alexander Engelsberger
2021-06-11 16:09:51 +02:00
parent 47d7f5831f
commit aaad2b8626
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/components/components.py
View File

@@ -118,7 +118,7 @@ class LabeledComponents(Components):
components, component_labels = parse_data_arg( components, component_labels = parse_data_arg(
initialized_components) initialized_components)
super().__init__(initialized_components=components) super().__init__(initialized_components=components)
self._labels = component_labels self._register_labels(component_labels)
else: else:
labels = get_labels_object(distribution) labels = get_labels_object(distribution)
self.initial_distribution = labels.distribution self.initial_distribution = labels.distribution