diff --git a/econtext/utils.py b/econtext/utils.py index 6b371e34..2e6bd331 100644 --- a/econtext/utils.py +++ b/econtext/utils.py @@ -2,13 +2,16 @@ import econtext +def run_with_broker(func, *args, **kwargs): + broker = econtext.Broker() + try: + return func(broker, *args, **kwargs) + finally: + broker.Finalize() + + def with_broker(func): def wrapper(*args, **kwargs): - broker = econtext.Broker() - try: - return func(broker, *args, **kwargs) - finally: - broker.Finalize() - + return run_with_broker(*args, **kwargs) wrapper.func_name = func.func_name return wrapper