diff options
Diffstat (limited to 'python/new_index_series.py')
| -rw-r--r-- | python/new_index_series.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/python/new_index_series.py b/python/new_index_series.py index 8c1e3d24..217463fd 100644 --- a/python/new_index_series.py +++ b/python/new_index_series.py @@ -1,5 +1,5 @@ -from bbg_helpers import BBG_IP, init_bbg_session, retrieve_data -from analytics.utils import roll_date +from serenitas.analytics.bbg_helpers import BBG_IP, init_bbg_session, retrieve_data +from serenitas.analytics.utils import roll_date from psycopg2.extras import execute_values import datetime @@ -81,7 +81,12 @@ def new_index_series(conn, basketid, df): c.executemany( "INSERT INTO basket_constituents VALUES(%s, %s, %s, %s)", [ - (t.company_id, t.seniority, basketid, 1.0,) + ( + t.company_id, + t.seniority, + basketid, + 1.0, + ) for t in df.itertuples(index=False) ], ) |