diff --git a/agent/base/control.py b/agent/base/control.py index 7889dd0..7e529f9 100644 --- a/agent/base/control.py +++ b/agent/base/control.py @@ -51,8 +51,10 @@ class SampleController(Controller): jobNumber = 0 + agents = (('sample01', 'base.sample'),) + def _getAgents(self): - return [AgentSpecification('sample01', 'base.sample')] + return [AgentSpecification(name, type) for name, type in self.agents] def enterJob(self, jobType, agent): self.jobNumber += 1 diff --git a/agent/core/control.py b/agent/core/control.py index 7078630..432d439 100644 --- a/agent/core/control.py +++ b/agent/core/control.py @@ -31,7 +31,6 @@ from cybertools.agent.components import controllers class SampleController(SampleController): - def _getAgents(self): - return [AgentSpecification('sample01', 'core.sample')] + agents = (('sample01', 'core.sample'),) controllers.register(SampleController, Master, name='core.sample')