Skip to content

Commit 4facab6

Browse files
committed
More precise type-hint for vectorize_graph
1 parent e25e8a2 commit 4facab6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytensor/graph/replace.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -232,13 +232,13 @@ def vectorize_graph(
232232
def vectorize_graph(
233233
outputs: Sequence[Variable],
234234
replace: Mapping[Variable, Variable],
235-
) -> Sequence[Variable]: ...
235+
) -> list[Variable]: ...
236236

237237

238238
def vectorize_graph(
239239
outputs: Variable | Sequence[Variable],
240240
replace: Mapping[Variable, Variable],
241-
) -> Variable | Sequence[Variable]:
241+
) -> Variable | list[Variable]:
242242
"""Vectorize outputs graph given mapping from old variables to expanded counterparts version.
243243
244244
Expanded dimensions must be on the left. Behavior is similar to the functional `numpy.vectorize`.

0 commit comments

Comments
 (0)