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():