From ca3f5665baa6e643b812d22ea9298579ea0093df Mon Sep 17 00:00:00 2001
From: Trenton H <797416+stumpylog@users.noreply.github.com>
Date: Sat, 11 Apr 2026 13:29:48 -0700
Subject: [PATCH] fix(profiling): correct docstring import path and add
 Callable type annotation

---
 profiling.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/profiling.py b/profiling.py
index 0d3c606f1..77beb6e44 100644
--- a/profiling.py
+++ b/profiling.py
@@ -3,7 +3,7 @@ Temporary profiling utilities for comparing implementations.
 
 Usage in a management command or shell::
 
-    from documents.profiling import profile_block
+    from profiling import profile_block, profile_cpu, measure_memory
 
     with profile_block("new check_sanity"):
         messages = check_sanity()
@@ -17,14 +17,12 @@ Drop this file when done.
 from __future__ import annotations
 
 import tracemalloc
+from collections.abc import Callable  # noqa: TC003
+from collections.abc import Generator  # noqa: TC003
 from contextlib import contextmanager
 from time import perf_counter
-from typing import TYPE_CHECKING
 from typing import Any
 
-if TYPE_CHECKING:
-    from collections.abc import Generator
-
 from django.db import connection
 from django.db import reset_queries
 from django.test.utils import override_settings
@@ -73,7 +71,7 @@ def profile_block(label: str = "block") -> Generator[None, None, None]:
 
 
 def profile_cpu(
-    fn,
+    fn: Callable[[], Any],
     *,
     label: str,
     top: int = 30,
@@ -114,7 +112,7 @@ def profile_cpu(
     return result, elapsed
 
 
-def measure_memory(fn, *, label: str) -> tuple[Any, float, float]:
+def measure_memory(fn: Callable[[], Any], *, label: str) -> tuple[Any, float, float]:
     """Run *fn()* under tracemalloc, print allocation report.
 
     Args: