From 400cbd1836e64bb51287587c4c1077cb7fba0250 Mon Sep 17 00:00:00 2001 From: Matthias Date: Fri, 12 May 2023 19:47:53 +0200 Subject: [PATCH] Fix types --- freqtrade/freqai/tensorboard/__init__.py | 4 ++-- tests/freqai/test_freqai_interface.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/freqtrade/freqai/tensorboard/__init__.py b/freqtrade/freqai/tensorboard/__init__.py index 14353de98..59862bc0d 100644 --- a/freqtrade/freqai/tensorboard/__init__.py +++ b/freqtrade/freqai/tensorboard/__init__.py @@ -6,8 +6,8 @@ try: except ModuleNotFoundError: from freqtrade.freqai.tensorboard.base_tensorboard import (BaseTensorBoardCallback, BaseTensorboardLogger) - TBLogger = BaseTensorboardLogger - TBCallback = BaseTensorBoardCallback + TBLogger = BaseTensorboardLogger # type: ignore + TBCallback = BaseTensorBoardCallback # type: ignore __all__ = ( "TBLogger", diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index 7e74fe5ed..ad615a5af 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -43,7 +43,7 @@ def can_run_model(model: str) -> None: @pytest.fixture(scope="function") -def import_fails_mac() -> None: +def import_fails_mac(): # Source of this test-method: # https://stackoverflow.com/questions/2481511/mocking-importerror-in-python if is_mac():