fix ruff issue

resgroup · Sep 13, 2024 · 9682a14 · 9682a14
1 parent 329ba8f
commit 9682a14
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_detrend.py b/tests/test_detrend.py
@@ -3,6 +3,7 @@
 import pandas as pd
 import pytest
 from pandas.testing import assert_frame_equal
+from pytest_benchmark.fixture import BenchmarkFixture
 
 from wind_up.detrend import apply_wsratio_v_wd_scen, calc_wsratio_v_wd_scen, check_applied_detrend
 from wind_up.models import WindUpConfig
@@ -66,7 +67,7 @@ def test_check_applied_detrend(test_lsa_t13_config: WindUpConfig) -> None:
     assert detrend_post_r2_improvement == pytest.approx(0.03776561982402227)
 
 
-def test_calc_wsratio_v_wd_scen(benchmark, test_lsa_t13_config: WindUpConfig) -> None:
+def test_calc_wsratio_v_wd_scen(benchmark: BenchmarkFixture, test_lsa_t13_config: WindUpConfig) -> None:
     # this test case borrows logic and results from check_applied_detrend where data which has already been detrended
     # is used to calculate the wsratio_v_wd_scen again to check it is flat
     cfg = test_lsa_t13_config