Dont reset pair priority if it doesnt successfully train

This commit is contained in:
robcaulk 2022-05-28 18:34:26 +02:00
parent 83dd453723
commit 4eb29c8810

View File

@ -325,7 +325,7 @@ class IFreqaiModel(ABC):
metadata) metadata)
except Exception: except Exception:
logger.warning('Mismatched sizes encountered in strategy') logger.warning('Mismatched sizes encountered in strategy')
self.data_drawer.pair_to_end_of_training_queue(metadata['pair']) # self.data_drawer.pair_to_end_of_training_queue(metadata['pair'])
self.training_on_separate_thread = False self.training_on_separate_thread = False
self.retrain = False self.retrain = False
return return
@ -334,7 +334,7 @@ class IFreqaiModel(ABC):
model = self.train(unfiltered_dataframe, metadata, dh) model = self.train(unfiltered_dataframe, metadata, dh)
except ValueError: except ValueError:
logger.warning('Value error encountered during training') logger.warning('Value error encountered during training')
self.data_drawer.pair_to_end_of_training_queue(metadata['pair']) # self.data_drawer.pair_to_end_of_training_queue(metadata['pair'])
self.training_on_separate_thread = False self.training_on_separate_thread = False
self.retrain = False self.retrain = False
return return