def __call__(self, control):
self.control = control<EOL>self.behaviors.add(control)<EOL>@wraps(control)<EOL>def wrapper(*args, **kwargs):<EOL><INDENT>return self.run(*args, **kwargs)<EOL><DEDENT>wrapper.experiment = self<EOL>return wrapper<EOL>
Wrap a control function.