diff --git a/pypsdm/models/result/participant/dict.py b/pypsdm/models/result/participant/dict.py index fa1d98e0..dc6bd9d4 100644 --- a/pypsdm/models/result/participant/dict.py +++ b/pypsdm/models/result/participant/dict.py @@ -25,6 +25,8 @@ class EntitiesResultDictMixin: + def uuids(self) -> set[str]: + return {key.uuid for key in self.keys()} # type: ignore @classmethod @abstractmethod @@ -142,7 +144,6 @@ def from_csv_for_entity( class EmsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin): - def __init__(self, data: dict[EntityKey, ComplexPower]): for key, value in data.items(): if not isinstance(key, EntityKey): @@ -167,7 +168,6 @@ def entity_type(cls) -> EntitiesEnum: class LoadsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin): - def __init__(self, data: dict[EntityKey, ComplexPower]): for key, value in data.items(): if not isinstance(key, EntityKey): @@ -192,7 +192,6 @@ def entity_type(cls) -> EntitiesEnum: class FixedFeedInsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin): - def __init__(self, data: dict[EntityKey, ComplexPower]): for key, value in data.items(): if not isinstance(key, EntityKey): @@ -217,7 +216,6 @@ def entity_type(cls) -> EntitiesEnum: class PvsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin): - def __init__(self, data: dict[EntityKey, ComplexPower]): for key, value in data.items(): if not isinstance(key, EntityKey): @@ -242,7 +240,6 @@ def entity_type(cls) -> EntitiesEnum: class WecsResult(ComplexPowerDict[EntityKey], EntitiesResultDictMixin): - def __init__(self, data: dict[EntityKey, ComplexPower]): for key, value in data.items(): if not isinstance(key, EntityKey): @@ -267,7 +264,6 @@ def entity_type(cls) -> EntitiesEnum: class StoragesResult(ComplexPowerWithSocDict[EntityKey], EntitiesResultDictMixin): - def __init__(self, data: dict[EntityKey, ComplexPowerWithSoc]): for key, value in data.items(): if not isinstance(key, EntityKey):