added functools.wraps as per @richafrank

This commit is contained in:
fawce
2013-02-19 13:15:57 -05:00
parent 9cc043f130
commit 5587c1bc64
+2
View File
@@ -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)