From 85a082a2ab96792aabd5bf7d178ee02824aff63a Mon Sep 17 00:00:00 2001 From: Austin Orr Date: Mon, 13 Dec 2021 23:40:24 -0800 Subject: [PATCH] appease linter --- lyra/lyra/src/viz/diversion_scenario.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/lyra/lyra/src/viz/diversion_scenario.py b/lyra/lyra/src/viz/diversion_scenario.py index bb400fc..7ec2574 100644 --- a/lyra/lyra/src/viz/diversion_scenario.py +++ b/lyra/lyra/src/viz/diversion_scenario.py @@ -27,8 +27,12 @@ def make_source_csv(source: pandas.DataFrame) -> str: def make_source(**kwargs: Dict) -> pandas.DataFrame: - df = simulate_diversion(**kwargs) # type: ignore - df: pandas.DataFrame = df.reset_index().melt(id_vars="date").round(4) + df = ( + simulate_diversion(**kwargs) # type: ignore + .reset_index() + .melt(id_vars="date") + .round(4) + ) return df