Update type hint for toposort_coords dictionary

ricardoV94 · web-flow · commit be4eb618a521 · 2025-11-25T17:40:42.000+01:00
diff --git a/pytensor/link/numba/dispatch/basic.py b/pytensor/link/numba/dispatch/basic.py
@@ -531,7 +531,7 @@ def numba_funcify_FunctionGraph(
     fgraph_can_be_cached = [True]
     cache_keys = []
     toposort = fgraph.toposort()
-    toposort_coords: dict[Variable, tuple[int, int]] = {
+    toposort_coords: dict[Variable, tuple[int, int | str]] = {
         inp: (0, i) for i, inp in enumerate(fgraph.inputs)
     }
     toposort_coords |= {

Original file line number	Diff line number	Diff line change
`@@ -531,7 +531,7 @@ def numba_funcify_FunctionGraph(`
`531`	`531`	`fgraph_can_be_cached = [True]`
`532`	`532`	`cache_keys = []`
`533`	`533`	`toposort = fgraph.toposort()`
`534`		`- toposort_coords: dict[Variable, tuple[int, int]] = {`
	`534`	`+ toposort_coords: dict[Variable, tuple[int, int \| str]] = {`
`535`	`535`	`inp: (0, i) for i, inp in enumerate(fgraph.inputs)`
`536`	`536`	`}`
`537`	`537`	`toposort_coords \|= {`