diff --git a/vessim/actor.py b/vessim/actor.py index c489aac..475976a 100644 --- a/vessim/actor.py +++ b/vessim/actor.py @@ -49,6 +49,9 @@ def state(self, now: datetime) -> dict: "p": self.p(now), } + def finalize(self) -> None: + self.signal.finalize() + class ComputingSystem(ActorBase): """Model of the computing system. @@ -89,6 +92,10 @@ def state(self, now: datetime) -> dict: "nodes": {signal.name: -signal.now(at=now) for signal in self.nodes}, } + def finalize(self) -> None: + for node in self.nodes: + node.finalize() + class _ActorSim(mosaik_api_v3.Simulator): META = { diff --git a/vessim/cosim.py b/vessim/cosim.py index c0765fa..d6d7131 100644 --- a/vessim/cosim.py +++ b/vessim/cosim.py @@ -102,6 +102,8 @@ def finalize(self): """ for controller in self.controllers: controller.finalize() + for actor in self.actors: + actor.finalize() class Environment: