make default type for close_price and date_pred np.nan

This commit is contained in:
Robert Caulk 2022-10-03 11:58:22 +02:00 committed by GitHub
parent 6ecd92de4a
commit 265795824b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -297,8 +297,8 @@ class FreqaiDataDrawer:
# 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'] = np.nan
df['date_pred'] = np.nan
df['close_price'].iloc[-1] = strat_df['close'].iloc[-1]
df['date_pred'].iloc[-1] = strat_df['date'].iloc[-1]