Automatically detect and change follower data_path to accommodate remote systems

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

View File

@ -154,6 +154,11 @@ class FreqaiDataKitchen:
if self.live:
self.model_filename = self.data_drawer.pair_dict[coin]['model_filename']
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:
self.data = json.load(fp)

View File

@ -191,7 +191,7 @@ class IFreqaiModel(ABC):
"""
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()
(model_filename,
@ -201,7 +201,7 @@ class IFreqaiModel(ABC):
# if the files do not yet exist, the follower returns null arrays to strategy
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)
return dh