Skip to content

Commit 223db70

Browse files
committed
Fixing CI failure
1 parent a4d3999 commit 223db70

File tree

1 file changed

+1
-1
lines changed
  • torch_geometric/nn/pool/select

1 file changed

+1
-1
lines changed

torch_geometric/nn/pool/select/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def __init__(
6262
self.weight = weight
6363

6464

65-
SelectOutput = torch.jit.script(SelectOutput)
65+
SelectOutput = torch.compile(SelectOutput)
6666

6767

6868
class Select(torch.nn.Module):

0 commit comments

Comments
 (0)