mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-21 12:30:16 +08:00
added functools.wraps as per @richafrank
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user