Skip to content

Commit dc2e79a

Browse files
committed
Improve results dataframe format
Default to_json() for MultiIndex is very large and hard to reconstruct
1 parent f0454ed commit dc2e79a

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

task_queue/tasks.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -88,9 +88,8 @@ def __run_simulation(simulation_input):
8888
skip_infer_datapackage_metadata=True,
8989
)
9090
logger.info(f"Simulation of {scenario} finished")
91-
df = calculator.df_results
92-
93-
simulation_output["results"] = df.to_json()
91+
results = {"flows": calculator.df_results.to_json(orient="split"), "kpis": calculator.kpis.to_json(orient="split")}
92+
simulation_output["results"] = results
9493
except Exception as e:
9594
logger.error(
9695
"An exception occured in the simulation task: {}".format(

0 commit comments

Comments
 (0)