revert to parquet capital

robcarver17 · Nov 17, 2023 · 5f14536 · 5f14536
1 parent 00b6cb6
commit 5f14536
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)