move data_provider cleanup to shutdown()

This commit is contained in:
robcaulk 2022-11-24 19:07:38 +01:00
parent 73c458d47b
commit 2e82e6784a

View File

@ -168,7 +168,6 @@ class IFreqaiModel(ABC):
""" """
self.model = None self.model = None
self.dk = None self.dk = None
self.data_provider = None
def _on_stop(self): def _on_stop(self):
""" """
@ -185,6 +184,7 @@ class IFreqaiModel(ABC):
logger.info("Stopping FreqAI") logger.info("Stopping FreqAI")
self._stop_event.set() self._stop_event.set()
self.data_provider = None
self._on_stop() self._on_stop()
logger.info("Waiting on Training iteration") logger.info("Waiting on Training iteration")