From 05d6913e040e1445aeda98b5b83dbcc165f4bc6a Mon Sep 17 00:00:00 2001 From: FilippoOlivo Date: Tue, 22 Apr 2025 13:25:04 +0200 Subject: [PATCH] Fix single graph handling --- pina/collector.py | 4 ---- pina/condition/condition_interface.py | 6 ++++++ 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/pina/collector.py b/pina/collector.py index db7296f..e0d04a0 100644 --- a/pina/collector.py +++ b/pina/collector.py @@ -104,10 +104,6 @@ class Collector: # get data keys = condition.__slots__ values = [getattr(condition, name) for name in keys] - values = [ - value.data if isinstance(value, Graph) else value - for value in values - ] self.data_collections[condition_name] = dict(zip(keys, values)) # condition now is ready self._is_conditions_ready[condition_name] = True diff --git a/pina/condition/condition_interface.py b/pina/condition/condition_interface.py index 9869c1e..ee20845 100644 --- a/pina/condition/condition_interface.py +++ b/pina/condition/condition_interface.py @@ -115,3 +115,9 @@ class ConditionInterface(metaclass=ABCMeta): raise ValueError( "LabelTensor must have the same labels" ) + + def __getattribute__(self, name): + to_return = super().__getattribute__(name) + if isinstance(to_return, (Graph, Data)): + to_return = [to_return] + return to_return