diff options
Diffstat (limited to 'python/yieldcurve.py')
| -rw-r--r-- | python/yieldcurve.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/yieldcurve.py b/python/yieldcurve.py index 230a7acd..997c7ad0 100644 --- a/python/yieldcurve.py +++ b/python/yieldcurve.py @@ -18,7 +18,7 @@ from quantlib.time.date import pydate_from_qldate import numpy as np from quantlib.quotes import SimpleQuote -from db import serenitas_engine +from db import dbconn, serenitas_engine from pyisda.curve import YieldCurve from pyisda.date import BadDay import warnings @@ -30,7 +30,8 @@ def load_curves(currency="USD", date=None): if date: sql_str += " WHERE effective_date=%s" - with closing(serenitas_engine.raw_connection()) as conn: + + with closing(dbconn('serenitasdb')) as conn: with conn.cursor() as c: if date: c.execute(sql_str, (date,)) |