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

Automatically detect and change follower data_path to accommodate remote systems

Browse Source
This commit is contained in:
robcaulk 2022-05-31 12:35:09 +02:00
parent 0306f5ca13
commit 70adf55643
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
freqtrade/freqai/data_kitchen.py
View File

@ -154,6 +154,11 @@ class FreqaiDataKitchen:
if self.live: if self.live:
self.model_filename = self.data_drawer.pair_dict[coin]['model_filename'] self.model_filename = self.data_drawer.pair_dict[coin]['model_filename']
self.data_path = Path(self.data_drawer.pair_dict[coin]['data_path']) self.data_path = Path(self.data_drawer.pair_dict[coin]['data_path'])
if self.freqai_config.get('follow_mode', False):
# follower can be on a different system which is rsynced to the leader:
self.data_path = Path(self.config["user_data_dir"] /
"models" / self.data_path.parts[-2] /
self.data_path.parts[-1])
with open(self.data_path / str(self.model_filename + "_metadata.json"), "r") as fp: with open(self.data_path / str(self.model_filename + "_metadata.json"), "r") as fp:
self.data = json.load(fp) self.data = json.load(fp)

4
freqtrade/freqai/freqai_interface.py
View File

@ -191,7 +191,7 @@ class IFreqaiModel(ABC):
""" """
if self.follow_mode: if self.follow_mode:
# follower needs to load off disk to get any changes made by leader to pair_dict # follower needs to load from disk to get any changes made by leader to pair_dict
self.data_drawer.load_drawer_from_disk() self.data_drawer.load_drawer_from_disk()
(model_filename, (model_filename,
@ -201,7 +201,7 @@ class IFreqaiModel(ABC):
# if the files do not yet exist, the follower returns null arrays to strategy # if the files do not yet exist, the follower returns null arrays to strategy
if self.follow_mode and return_null_array: if self.follow_mode and return_null_array:
logger.info('Returning null array from follower') logger.info('Returning null array from follower to strategy')
self.data_drawer.return_null_values_to_strategy(dataframe, dh) self.data_drawer.return_null_values_to_strategy(dataframe, dh)
return dh return dh