diff options
Diffstat (limited to 'python/notify_novations.py')
| -rw-r--r-- | python/notify_novations.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/python/notify_novations.py b/python/notify_novations.py index a4ce00e7..e07c864f 100644 --- a/python/notify_novations.py +++ b/python/notify_novations.py @@ -87,9 +87,9 @@ def contact_novation_counterparties(date): & (df["ThirdPartyStatus.1"].isin(["Pending Consent"])) ] df = ( - df.groupby(["ExecutingBroker", "RemainingBroker", "CurrencyCode"])[ - "Novation Consent ID" - ] + df.groupby( + ["ExecutingBroker", "RemainingBroker", "CurrencyCode"], group_keys=False + )["Novation Consent ID"] .apply(list) .reset_index() ) @@ -116,6 +116,7 @@ if __name__ == "__main__": "date", type=datetime.date.fromisoformat, default=datetime.date.today(), + nargs="?", ) args = parser.parse_args() contact_novation_counterparties(args.date) |