diff options
Diffstat (limited to 'python/analytics/utils.py')
| -rw-r--r-- | python/analytics/utils.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/python/analytics/utils.py b/python/analytics/utils.py index 885c4db8..750f4488 100644 --- a/python/analytics/utils.py +++ b/python/analytics/utils.py @@ -1,3 +1,4 @@ +import analytics import datetime import numpy as np import pandas as pd @@ -5,6 +6,7 @@ from . import dbconn from .exceptions import MissingDataError from scipy.special import h_roots from dateutil.relativedelta import relativedelta, WE +from contextlib import contextmanager from functools import partial, wraps, lru_cache from pyisda.date import pydate_to_TDate from pandas.api.types import CategoricalDtype @@ -219,3 +221,11 @@ def get_fx(value_date: datetime.date, currency: str): ) conn.close() return r + + +@contextmanager +def run_local(local=True): + saved_local = analytics._local + analytics._local = local + yield + analytics._local = saved_local |