Mock tensorboard callbacks for all freqAI tests

This commit is contained in:
Matthias
2023-05-13 08:02:19 +02:00
parent 23e8932a44
commit ab0f9d78ee
2 changed files with 15 additions and 10 deletions
+14
View File
@@ -1,3 +1,4 @@
import platform
from copy import deepcopy
from pathlib import Path
from typing import Any, Dict
@@ -14,6 +15,19 @@ from freqtrade.resolvers.freqaimodel_resolver import FreqaiModelResolver
from tests.conftest import get_patched_exchange
def is_mac() -> bool:
machine = platform.system()
return "Darwin" in machine
@pytest.fixture(scope="function", autouse=True)
def patch_tensorboard_maconly(mocker):
if is_mac():
# MacOS CI is not friendly to tensorboard
mocker.patch('freqtrade.freqai.tensorboard.tensorboard.SummaryWriter')
mocker.patch('freqtrade.freqai.tensorboard.tensorboard.TensorBoardCallback.after_iteration')
@pytest.fixture(scope="function")
def freqai_conf(default_conf, tmpdir):
freqaiconf = deepcopy(default_conf)