Skip to content

Commit 0bf00ab

Browse files
author
Ian Schweer
committed
Fix linter
1 parent 738393f commit 0bf00ab

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytensor/link/pytorch/dispatch/basic.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,10 @@ def ifelse(cond, *true_and_false, n_outs=n_outs):
163163

164164
return ifelse
165165

166+
166167
@pytorch_funcify.register(TensorFromScalar)
167168
def pytorch_funcify_TensorFromScalar(op, **kwargs):
168169
def tensorfromscalar(x):
169170
return torch.as_tensor(x)
170171

171-
return tensorfromscalar
172+
return tensorfromscalar

0 commit comments

Comments
 (0)