From cda869f04b9d5c78b1cd45c0fb1a21774a8b5b9a Mon Sep 17 00:00:00 2001 From: Chris Guidry Date: Mon, 29 Jul 2024 15:12:48 -0400 Subject: [PATCH] Just go back to auto --- benches/bench_import.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/benches/bench_import.py b/benches/bench_import.py index dbc1473c6b43..0398ec523ce0 100644 --- a/benches/bench_import.py +++ b/benches/bench_import.py @@ -23,15 +23,19 @@ def reset_imports(): @pytest.mark.benchmark(group="imports") def bench_import_prefect(benchmark: BenchmarkFixture): def import_prefect(): + reset_imports() + import prefect # noqa - benchmark.pedantic(import_prefect, setup=reset_imports, rounds=15) + benchmark(import_prefect) @pytest.mark.timeout(180) @pytest.mark.benchmark(group="imports") def bench_import_prefect_flow(benchmark: BenchmarkFixture): def import_prefect_flow(): + reset_imports() + from prefect import flow # noqa - benchmark.pedantic(import_prefect_flow, setup=reset_imports, rounds=15) + benchmark(import_prefect_flow)