From 5587c1bc64272248e287b66574ba3aa49ebb817c Mon Sep 17 00:00:00 2001
From: fawce <fawce@quantopian.com>
Date: Tue, 19 Feb 2013 13:15:57 -0500
Subject: [PATCH] added functools.wraps as per @richafrank

---
 zipline/finance/trading.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/zipline/finance/trading.py b/zipline/finance/trading.py
index 9bd7dc5a..c5ea3850 100644
--- a/zipline/finance/trading.py
+++ b/zipline/finance/trading.py
@@ -18,6 +18,7 @@ import pytz
 import logbook
 import datetime
 
+from functools import wraps
 from collections import defaultdict, OrderedDict
 from delorean import Delorean
 from pandas import DatetimeIndex
@@ -366,6 +367,7 @@ class use_environment(object):
         self.env = environment
 
     def __call__(self, func):
+        @wraps(func)
         def wrapper(*args, **kwargs):
             with self.env:
                 return func(*args, **kwargs)