def update_variables(func):
@wraps(func)<EOL>def wrapper(self, *args, **kwargs):<EOL><INDENT>result = func(self, *args, **kwargs)<EOL>if isinstance(result, tuple):<EOL><INDENT>return self.process_register(result[<NUM_LIT:0>], result[<NUM_LIT:1>])<EOL><DEDENT>else:<EOL><INDENT>return self.process_register(result)<EOL><DEDENT><DEDENT>return wrapper<EOL>
Use this decorator on Step.action implementation.

Your action method should always return variables, or
both variables and output.

This decorator will update variables with output.