Skip to content

Commit d00f9e2

Browse files
author
Ian Schweer
committed
Lint
1 parent 46e3e72 commit d00f9e2

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

pytensor/link/pytorch/dispatch/elemwise.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import importlib
2+
23
import torch
34

45
from pytensor.link.pytorch.dispatch.basic import pytorch_funcify

pytensor/link/pytorch/dispatch/scalar.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
Cast,
99
ScalarOp,
1010
)
11-
from pytensor.scalar.math import Softplus
1211
from pytensor.scalar.loop import ScalarLoop
12+
from pytensor.scalar.math import Softplus
1313

1414

1515
@pytorch_funcify.register(ScalarOp)
@@ -60,6 +60,7 @@ def cast(x):
6060

6161
return cast
6262

63+
6364
@pytorch_funcify.register(Softplus)
6465
def pytorch_funcify_Softplus(op, node, **kwargs):
6566
return torch.nn.Softplus()

0 commit comments

Comments
 (0)