fix the typing for the size

williambdean · williambdean · commit 04337581e618 · 2025-08-03T01:09:25.000-04:00
diff --git a/pymc/testing.py b/pymc/testing.py
@@ -978,7 +978,7 @@ def assert_no_rvs(vars: Sequence[Variable]) -> None:
         raise AssertionError(f"RV found in graph: {rvs}")
 
 
-SampleStatsCreator = Callable[[tuple[str, ...]], NDArray]
+SampleStatsCreator = Callable[[tuple[int, int]], NDArray]
 
 
 def mock_sample(
