fix test_start_backtesting

This commit is contained in:
Yinon Polak 2023-03-21 16:38:05 +02:00
parent 4f93106755
commit b9c7d338b3
2 changed files with 4 additions and 2 deletions

View File

@ -1,5 +1,6 @@
from copy import deepcopy
from pathlib import Path
from typing import Any, Dict
from unittest.mock import MagicMock
import pytest
@ -83,7 +84,7 @@ def make_rl_config(conf):
return conf
def mock_pytorch_mlp_model_training_parameters(conf):
def mock_pytorch_mlp_model_training_parameters() -> Dict[str, Any]:
return {
"learning_rate": 3e-4,
"trainer_kwargs": {

View File

@ -15,7 +15,8 @@ from freqtrade.optimize.backtesting import Backtesting
from freqtrade.persistence import Trade
from freqtrade.plugins.pairlistmanager import PairListManager
from tests.conftest import EXMS, create_mock_trades, get_patched_exchange, log_has_re
from tests.freqai.conftest import get_patched_freqai_strategy, make_rl_config
from tests.freqai.conftest import get_patched_freqai_strategy, make_rl_config, \
mock_pytorch_mlp_model_training_parameters
def is_py11() -> bool: