From c8ecedf6d56f8f3b22669401f0c3224aa26938d0 Mon Sep 17 00:00:00 2001 From: Matthias Date: Tue, 17 Jan 2023 20:05:18 +0100 Subject: [PATCH] Clarify a variable via typehint --- freqtrade/optimize/backtesting.py | 4 ++-- tests/strategy/test_strategy_loading.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/freqtrade/optimize/backtesting.py b/freqtrade/optimize/backtesting.py index a01884ae0..2391605bc 100644 --- a/freqtrade/optimize/backtesting.py +++ b/freqtrade/optimize/backtesting.py @@ -15,7 +15,7 @@ from pandas import DataFrame from freqtrade import constants from freqtrade.configuration import TimeRange, validate_config_consistency -from freqtrade.constants import DATETIME_PRINT_FORMAT, Config, LongShort +from freqtrade.constants import DATETIME_PRINT_FORMAT, Config, IntOrInf, LongShort from freqtrade.data import history from freqtrade.data.btanalysis import find_existing_backtest_stats, trade_list_to_dataframe from freqtrade.data.converter import trim_dataframe, trim_dataframes @@ -924,7 +924,7 @@ class Backtesting: def trade_slot_available(self, open_trade_count: int) -> bool: # Always allow trades when max_open_trades is enabled. - max_open_trades = self.config['max_open_trades'] + max_open_trades: IntOrInf = self.config['max_open_trades'] if max_open_trades <= 0 or open_trade_count < max_open_trades: return True # Rejected trade diff --git a/tests/strategy/test_strategy_loading.py b/tests/strategy/test_strategy_loading.py index d60d3ade9..98185e152 100644 --- a/tests/strategy/test_strategy_loading.py +++ b/tests/strategy/test_strategy_loading.py @@ -6,7 +6,7 @@ from pathlib import Path import pytest from pandas import DataFrame -from freqtrade.configuration.configuration import Configuration +from freqtrade.configuration import Configuration from freqtrade.exceptions import OperationalException from freqtrade.resolvers import StrategyResolver from freqtrade.strategy.interface import IStrategy