undo to pickle without float32 casting

This commit is contained in:
longyu 2023-01-14 08:58:15 +01:00
parent 145d9fd633
commit 6d8267328f

View File

@ -518,7 +518,7 @@ class FreqaiDataDrawer:
rapidjson.dump(dk.data, fp, default=self.np_encoder, number_mode=rapidjson.NM_NATIVE)
# save the train data to file so we can check preds for area of applicability later
dk.data_dictionary["train_features"].astype("float32").to_pickle(
dk.data_dictionary["train_features"].to_pickle(
save_path / f"{dk.model_filename}_trained_df.pkl.bz2"
)