adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reduce duplicated code

Browse Source
This commit is contained in:
robcaulk 2023-03-25 13:27:41 +01:00
parent b383654378
commit 630cdcb99f
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
freqtrade/freqai/data_drawer.py
View File

@ -169,11 +169,7 @@ class FreqaiDataDrawer:
if exists:
try:
for file_path in self.historic_predictions_folder.glob("*.parquet"):
key = file_path.stem
key.replace("_", "/")
self.historic_predictions[key] = pd.read_parquet(file_path)
self.load_historic_predictions_from_folder()
logger.info(
f"Found existing historic predictions at {self.full_path}, but beware "
"that statistics may be inaccurate if the bot has been offline for "
@ -182,10 +178,7 @@ class FreqaiDataDrawer:
except EOFError:
logger.warning(
'Historical prediction files were corrupted. Trying to load backup files.')
for file_path in self.historic_predictions_folder.glob("*.parquet"):
key = file_path.stem
key.replace("_", "/")
self.historic_predictions[key] = pd.read_parquet(file_path)
self.load_historic_predictions_from_folder()
logger.warning('FreqAI successfully loaded the backup '
'historical predictions files.')
@ -205,6 +198,18 @@ class FreqaiDataDrawer:
return exists
def load_historic_predictions_from_folder(self):
"""
Try to build the historic_predictions dictionary from parquet
files in the historic_predictions_folder
"""
for file_path in self.historic_predictions_folder.glob("*.parquet"):
key = file_path.stem
key.replace("_", "/")
self.historic_predictions[key] = pd.read_parquet(file_path)
return
def save_historic_predictions_to_disk(self):
"""
Save historic predictions pickle to disk