diff --git a/pfdl_scheduler/model/service.py b/pfdl_scheduler/model/service.py index 2ab157c..9de00f7 100644 --- a/pfdl_scheduler/model/service.py +++ b/pfdl_scheduler/model/service.py @@ -19,7 +19,6 @@ from pfdl_scheduler.model.array import Array -@dataclass class Service: """Represents a Service or Service Call in the PFDL. diff --git a/pfdl_scheduler/petri_net/generator.py b/pfdl_scheduler/petri_net/generator.py index 0d39864..3333228 100644 --- a/pfdl_scheduler/petri_net/generator.py +++ b/pfdl_scheduler/petri_net/generator.py @@ -127,6 +127,7 @@ def __init__( self.tree = None self.file_name = file_name self.pfdl_base_classes = pfdl_base_classes + self.service_apis: list[ServiceAPI] = [] def add_callback(self, transition_uuid: str, callback_function: Callable, *args: Any) -> None: """Registers the given callback function in the transition_dict. @@ -288,6 +289,7 @@ def generate_service( service_api = self.pfdl_base_classes.get_class("ServiceAPI")( service, task_context, in_loop=in_loop ) + self.service_apis.append(service_api) service_started_uuid = create_place(service.name + " started", self.net, service_node) service_finished_uuid = create_place(service.name + " finished", self.net, service_node)