Skip to content

Commit 9ee0ee7

Browse files
committed
Fix wrong type-hint in indices_from_subtensor
1 parent a1a805c commit 9ee0ee7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/tensor/subtensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@
8585
def indices_from_subtensor(
8686
op_indices: Iterable[ScalarConstant],
8787
idx_list: Optional[List[Union[Type, slice, Variable]]],
88-
) -> Union[slice, Variable]:
88+
) -> Tuple[Union[slice, Variable], ...]:
8989
"""Recreate the index tuple from which a ``*Subtensor**`` ``Op`` was created.
9090
9191
Parameters

0 commit comments

Comments
 (0)