def revert_overrides_agents(self):
if not hasattr(self, 'overriden_agents'):
return
else:
for agent_type, factory in self.overriden_agents.iteritems():
if factory:
registry.override(agent_type, factory)
def override_agent(self, agent_type, factory):
if not hasattr(self, 'overriden_agents'):
self.overriden_agents = dict()
old = registry.registry_lookup(agent_type)
self.overriden_agents[agent_type] = old
registry.override(agent_type, factory)
