Allow updating without changing identifier

This commit is contained in:
Robert Caulk 2022-10-03 09:55:57 +02:00 committed by GitHub
parent b70f18f4c3
commit 6ecd92de4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions

View File

@ -294,6 +294,12 @@ class FreqaiDataDrawer:
for return_str in rets:
df[return_str].iloc[-1] = rets[return_str]
# this logic carries users between version without needing to
# change their identifier
if 'close_price' not in df.columns:
df['close_price'] = 0
df['date_pred'] = 0
df['close_price'].iloc[-1] = strat_df['close'].iloc[-1]
df['date_pred'].iloc[-1] = strat_df['date'].iloc[-1]