From 5f1453630e560df37823d78c68c0dec07923fecc Mon Sep 17 00:00:00 2001 From: Rob Carver Date: Fri, 17 Nov 2023 15:05:20 +0000 Subject: [PATCH] revert to parquet capital --- sysdata/parquet/parquet_capital.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sysdata/parquet/parquet_capital.py b/sysdata/parquet/parquet_capital.py index 0f9b6af0ed..1e9a794c93 100644 --- a/sysdata/parquet/parquet_capital.py +++ b/sysdata/parquet/parquet_capital.py @@ -45,5 +45,8 @@ def _delete_all_capital_for_strategy_no_checking(self, strategy_name: str): def update_capital_pd_df_for_strategy( self, strategy_name: str, updated_capital_df: pd.DataFrame ): - updated_capital_df.columns = ['capital'] + if len(updated_capital_df.columns)==1: + ## single strategy, need columns labelling + updated_capital_df.columns = ['capital'] + self.parquet.write_data_given_data_type_and_identifier(data_to_write=updated_capital_df, identifier=strategy_name, data_type=CAPITAL_COLLECTION)