ensure dates are saved

This commit is contained in:
robcaulk 2022-08-12 12:03:44 +02:00
parent 2cae3c42e6
commit fb4b73ce89

View File

@ -202,7 +202,7 @@ class FreqaiDataKitchen:
labels = unfiltered_dataframe.filter(label_list, axis=1) labels = unfiltered_dataframe.filter(label_list, axis=1)
drop_index_labels = pd.isnull(labels).any(1) drop_index_labels = pd.isnull(labels).any(1)
drop_index_labels = drop_index_labels.replace(True, 1).replace(False, 0) drop_index_labels = drop_index_labels.replace(True, 1).replace(False, 0)
dates = unfiltered_dataframe.filter('date', axis=1) dates = unfiltered_dataframe['date']
filtered_dataframe = filtered_dataframe[ filtered_dataframe = filtered_dataframe[
(drop_index == 0) & (drop_index_labels == 0) (drop_index == 0) & (drop_index_labels == 0)
] # dropping values ] # dropping values