Just go back to auto

PrefectHQ · Jul 29, 2024 · cda869f · cda869f
1 parent f438fec
commit cda869f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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)