Fix errors in markdown and codspeed workflows

.github/workflows/codspeed-benchmarks.yaml

@@ -61,7 +61,8 @@ jobs:
       - name: Install packages
         run: |
           python -m pip install -U uv
-          uv pip install --upgrade --system .[dev]
+          uv pip install --upgrade --system .[dev] pytest-codspeed
+          uv pip uninstall --system pytest-benchmark
 
       - name: Start server
         run: |

.github/workflows/markdown-tests.yaml

@@ -95,6 +95,7 @@ jobs:
           python -m pip install -U uv
           uv pip install --upgrade --system -e '.[dev]'
           uv pip install --upgrade --system -r requirements-markdown-tests.txt
+          uv pip uninstall --system pytest-benchmark
 
       - name: Start server
         run: |

benches/bench_flows.py

@@ -2,12 +2,16 @@
 TODO: Add benches for higher number of tasks; blocked by engine deadlocks in CI.
 """
 
+from typing import TYPE_CHECKING
+
 import anyio
 import pytest
-from pytest_benchmark.fixture import BenchmarkFixture
 
 from prefect import flow, task
 
+if TYPE_CHECKING:
+    from pytest_benchmark.fixture import BenchmarkFixture
+
 
 def noop_function():
     pass
@@ -17,12 +21,12 @@ async def anoop_function():
     pass
 
 
-def bench_flow_decorator(benchmark: BenchmarkFixture):
+def bench_flow_decorator(benchmark: "BenchmarkFixture"):
     benchmark(flow, noop_function)
 
 
 @pytest.mark.parametrize("options", [{}, {"timeout_seconds": 10}])
-def bench_flow_call(benchmark: BenchmarkFixture, options):
+def bench_flow_call(benchmark: "BenchmarkFixture", options):
     noop_flow = flow(**options)(noop_function)
     benchmark(noop_flow)
 
@@ -35,7 +39,7 @@ def bench_flow_call(benchmark: BenchmarkFixture, options):
 
 
 @pytest.mark.parametrize("num_tasks", [10, 50, 100])
-def bench_flow_with_submitted_tasks(benchmark: BenchmarkFixture, num_tasks: int):
+def bench_flow_with_submitted_tasks(benchmark: "BenchmarkFixture", num_tasks: int):
     test_task = task(noop_function)
 
     @flow
@@ -47,7 +51,7 @@ def benchmark_flow():
 
 
 @pytest.mark.parametrize("num_tasks", [10, 50, 100, 250])
-def bench_flow_with_called_tasks(benchmark: BenchmarkFixture, num_tasks: int):
+def bench_flow_with_called_tasks(benchmark: "BenchmarkFixture", num_tasks: int):
     test_task = task(noop_function)
 
     @flow
@@ -62,7 +66,7 @@ def benchmark_flow():
 
 
 @pytest.mark.parametrize("num_tasks", [10, 50, 100, 250])
-def bench_async_flow_with_async_tasks(benchmark: BenchmarkFixture, num_tasks: int):
+def bench_async_flow_with_async_tasks(benchmark: "BenchmarkFixture", num_tasks: int):
     test_task = task(anoop_function)
 
     @flow
@@ -78,7 +82,7 @@ async def benchmark_flow():
 
 
 @pytest.mark.parametrize("num_flows", [5, 10, 20])
-def bench_flow_with_subflows(benchmark: BenchmarkFixture, num_flows: int):
+def bench_flow_with_subflows(benchmark: "BenchmarkFixture", num_flows: int):
     test_flow = flow(noop_function)
 
     @flow
@@ -91,7 +95,7 @@ def benchmark_flow():
 
 @pytest.mark.parametrize("num_flows", [5, 10, 20])
 def bench_async_flow_with_sequential_subflows(
-    benchmark: BenchmarkFixture, num_flows: int
+    benchmark: "BenchmarkFixture", num_flows: int
 ):
     test_flow = flow(anoop_function)
 
@@ -105,7 +109,7 @@ async def benchmark_flow():
 
 @pytest.mark.parametrize("num_flows", [5, 10, 20])
 def bench_async_flow_with_concurrent_subflows(
-    benchmark: BenchmarkFixture, num_flows: int
+    benchmark: "BenchmarkFixture", num_flows: int
 ):
     test_flow = flow(anoop_function)
 

benches/bench_import.py

@@ -1,9 +1,12 @@
 import importlib
 import sys
+from typing import TYPE_CHECKING
 
 import pytest
 from prometheus_client import REGISTRY
-from pytest_benchmark.fixture import BenchmarkFixture
+
+if TYPE_CHECKING:
+    from pytest_benchmark.fixture import BenchmarkFixture
 
 
 def reset_imports():
@@ -21,7 +24,7 @@ def reset_imports():
 
 
 @pytest.mark.benchmark(group="imports")
-def bench_import_prefect(benchmark: BenchmarkFixture):
+def bench_import_prefect(benchmark: "BenchmarkFixture"):
     def import_prefect():
         reset_imports()
 
@@ -32,7 +35,7 @@ def import_prefect():
 
 @pytest.mark.timeout(180)
 @pytest.mark.benchmark(group="imports")
-def bench_import_prefect_flow(benchmark: BenchmarkFixture):
+def bench_import_prefect_flow(benchmark: "BenchmarkFixture"):
     def import_prefect_flow():
         reset_imports()
 

benches/bench_tasks.py

@@ -1,4 +1,7 @@
-from pytest_benchmark.fixture import BenchmarkFixture
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pytest_benchmark.fixture import BenchmarkFixture
 
 from prefect import flow, task
 
@@ -7,11 +10,11 @@ def noop_function():
     pass
 
 
-def bench_task_decorator(benchmark: BenchmarkFixture):
+def bench_task_decorator(benchmark: "BenchmarkFixture"):
     benchmark(task, noop_function)
 
 
-def bench_task_call(benchmark: BenchmarkFixture):
+def bench_task_call(benchmark: "BenchmarkFixture"):
     noop_task = task(noop_function)
 
     @flow
@@ -21,7 +24,7 @@ def benchmark_flow():
     benchmark_flow()
 
 
-def bench_task_submit(benchmark: BenchmarkFixture):
+def bench_task_submit(benchmark: "BenchmarkFixture"):
     noop_task = task(noop_function)
 
     # The benchmark occurs within the flow to measure _submission_ time without

requirements-dev.txt

@@ -12,7 +12,6 @@ pluggy >= 1.4.0
 pytest >= 8.3
 pytest-asyncio >= 0.24
 pytest-benchmark
-pytest-codspeed
 pytest-cov
 pytest-env
 pytest-flakefinder