fix: allow user to break training

This commit is contained in:
Robert Caulk
2024-10-21 11:40:46 +02:00
parent 6571ef4e45
commit a1513b7c7c

View File

@@ -199,9 +199,16 @@ class IFreqaiModel(ABC):
self.data_provider = None
self._on_stop()
logger.info("Waiting on Training iteration")
for _thread in self._threads:
_thread.join()
if self.freqai_info.get("wait_for_training_iteration_on_reload", True):
logger.info("Waiting on Training iteration")
for _thread in self._threads:
_thread.join()
else:
logger.warning(
"Breaking current training iteration because "
"you set wait_for_training_iteration_on_reload to "
" False."
)
def start_scanning(self, *args, **kwargs) -> None:
"""