PickleGzInstance: Replace implicit load by load/free methods

This commit is contained in:
2021-04-06 19:19:12 -05:00
parent f495297168
commit 856b595d5e
6 changed files with 28 additions and 29 deletions

View File

@@ -104,10 +104,12 @@ class DynamicConstraintsComponent(Component):
def fit(self, training_instances: List["Instance"]) -> None:
collected_cids = set()
for instance in training_instances:
instance.load()
for sample in instance.training_data:
if getattr(sample, self.attr) is None:
continue
collected_cids |= getattr(sample, self.attr)
instance.free()
self.known_cids.clear()
self.known_cids.extend(sorted(collected_cids))
super().fit(training_instances)