From b01aaa4d035e195ab71af0001e257d53d68aa0e4 Mon Sep 17 00:00:00 2001 From: robcaulk Date: Wed, 10 May 2023 10:11:33 +0000 Subject: [PATCH] ensure backtesting also produces tb_logs, make sure tests are working --- freqtrade/freqai/freqai_interface.py | 2 ++ tests/freqai/test_freqai_datakitchen.py | 1 + tests/freqai/test_freqai_interface.py | 2 -- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/freqtrade/freqai/freqai_interface.py b/freqtrade/freqai/freqai_interface.py index b453500b8..3344f3df4 100644 --- a/freqtrade/freqai/freqai_interface.py +++ b/freqtrade/freqai/freqai_interface.py @@ -345,7 +345,9 @@ class IFreqaiModel(ABC): dk.find_labels(dataframe_train) try: + self.tb_logger = TBLogger(dk.data_path) self.model = self.train(dataframe_train, pair, dk) + self.tb_logger.close() except Exception as msg: logger.warning( f"Training {pair} raised exception {msg.__class__.__name__}. " diff --git a/tests/freqai/test_freqai_datakitchen.py b/tests/freqai/test_freqai_datakitchen.py index 3f0fc697d..c9d3a973c 100644 --- a/tests/freqai/test_freqai_datakitchen.py +++ b/tests/freqai/test_freqai_datakitchen.py @@ -194,3 +194,4 @@ def test_get_full_model_path(mocker, freqai_conf, model): model_path = freqai.dk.get_full_models_path(freqai_conf) assert model_path.is_dir() is True + shutil.rmtree(Path(freqai.dk.full_path)) diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index e27c8d2c0..5291185f0 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -552,8 +552,6 @@ def test_get_state_info(mocker, freqai_conf, dp_exists, caplog, tickers): if is_mac(): pytest.skip("Reinforcement learning module not available on intel based Mac OS") - if is_py11(): - pytest.skip("Reinforcement learning currently not available on python 3.11.") freqai_conf.update({"freqaimodel": "ReinforcementLearner"}) freqai_conf.update({"timerange": "20180110-20180130"})