From ebae0a7248ca37d3af8519bce45d1c5b3b84a2af Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Tue, 3 Dec 2024 19:44:26 +0100
Subject: [PATCH] chore: improve typing of new functionality

---
 freqtrade/optimize/optimize_reports/optimize_reports.py | 6 +++---
 freqtrade/util/dry_run_wallet.py                        | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/freqtrade/optimize/optimize_reports/optimize_reports.py b/freqtrade/optimize/optimize_reports/optimize_reports.py
index 532abed39..23119077b 100644
--- a/freqtrade/optimize/optimize_reports/optimize_reports.py
+++ b/freqtrade/optimize/optimize_reports/optimize_reports.py
@@ -69,7 +69,7 @@ def generate_rejected_signals(
 
 
 def _generate_result_line(
-    result: DataFrame, starting_balance: int, first_column: str | list[str]
+    result: DataFrame, starting_balance: float, first_column: str | list[str]
 ) -> dict:
     """
     Generate one result dict, with "first_column" as key.
@@ -111,7 +111,7 @@ def _generate_result_line(
 def generate_pair_metrics(
     pairlist: list[str],
     stake_currency: str,
-    starting_balance: int,
+    starting_balance: float,
     results: DataFrame,
     skip_nan: bool = False,
 ) -> list[dict]:
@@ -144,7 +144,7 @@ def generate_pair_metrics(
 
 def generate_tag_metrics(
     tag_type: Literal["enter_tag", "exit_reason"] | list[Literal["enter_tag", "exit_reason"]],
-    starting_balance: int,
+    starting_balance: float,
     results: DataFrame,
     skip_nan: bool = False,
 ) -> list[dict]:
diff --git a/freqtrade/util/dry_run_wallet.py b/freqtrade/util/dry_run_wallet.py
index f994f0d4c..c904db6b3 100644
--- a/freqtrade/util/dry_run_wallet.py
+++ b/freqtrade/util/dry_run_wallet.py
@@ -1,4 +1,4 @@
-from pytest import Config
+from freqtrade.constants import Config
 
 
 def get_dry_run_wallet(config: Config) -> int | float: