utils.run_with_broker.

econtext/utils.py

@@ -2,13 +2,16 @@
 import econtext
 
 
-def with_broker(func):
+def run_with_broker(func, *args, **kwargs):
-    def wrapper(*args, **kwargs):
     broker = econtext.Broker()
     try:
         return func(broker, *args, **kwargs)
     finally:
         broker.Finalize()
 
+
+def with_broker(func):
+    def wrapper(*args, **kwargs):
+        return run_with_broker(*args, **kwargs)
     wrapper.func_name = func.func_name
     return wrapper