diff options
| -rw-r--r-- | python/bbg_prices.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/python/bbg_prices.py b/python/bbg_prices.py index 384f449c..834aa9b1 100644 --- a/python/bbg_prices.py +++ b/python/bbg_prices.py @@ -3,11 +3,9 @@ from sqlalchemy import create_engine import numpy as np import pandas as pd -from psycopg2.extensions import adapt, register_adapter, AsIs -def adapt_nat(nat): - return AsIs(None) +from psycopg2.extensions import register_adapter, AsIs -register_adapter(pandas.tslib.NaTType, adapt_nat) +register_adapter(pd.tslib.NaTType, lambda nat: AsIs(None)) engine = create_engine('postgresql://et_user@debian/ET') |