diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/risk/__main__.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/python/risk/__main__.py b/python/risk/__main__.py index 475e5d9f..c530c695 100644 --- a/python/risk/__main__.py +++ b/python/risk/__main__.py @@ -1,6 +1,6 @@ import analytics import argparse -import pandas as pd +import datetime from . import dbconn, dbengine from dates import bus_day from .bonds import subprime_risk, clo_risk, crt_risk, insert_subprime_risk @@ -10,12 +10,10 @@ from .swaptions import get_swaption_portfolio, insert_swaption_portfolio from .tranches import get_tranche_portfolio, insert_tranche_portfolio parser = argparse.ArgumentParser() -parser.add_argument( - "workdate", nargs="?", type=lambda s: pd.datetime.strptime(s, "%Y-%m-%d").date() -), +parser.add_argument("workdate", nargs="?", type=datetime.date.fromisoformat), args = parser.parse_args() if args.workdate is None: - workdate = (pd.Timestamp.today() - bus_day).date() + workdate = (datetime.date.today() - bus_day).date() else: workdate = args.workdate |