Skip to content

Commit 2de9566

Browse files
committed
Picking a nit
1 parent f000bbb commit 2de9566

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/xtensor/rewriting/shape.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def lower_transpose(fgraph, node):
123123
@node_rewriter([Squeeze])
124124
def local_squeeze_reshape(fgraph, node):
125125
"""Rewrite Squeeze to tensor.squeeze."""
126-
x = node.inputs[0]
126+
[x] = node.inputs
127127
x_tensor = tensor_from_xtensor(x)
128128
x_dims = x.type.dims
129129
dims_to_remove = node.op.dims

0 commit comments

Comments
 (0)