[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

benchmark/ops/softmax.py

@@ -1,11 +1,11 @@
 import argparse
+from time import perf_counter as timestamp
 
 import torch
-import pyg_lib
-
-from time import perf_counter as timestamp
 from torch_geometric.utils import segment
 
+import pyg_lib
+
 
 def softmax_reference_ptr(src, ptr, dim=0):
     dim = dim + src.dim() if dim < 0 else dim
@@ -66,4 +66,3 @@ def measure_perf(impl_func, ptr, out_grad, num_warmups, num_steps, backward):
     print(f'pyg_lib forward:  {t_fwd:.4f}s')
     if args.backward:
         print(f'pyg_lib backward: {t_bwd:.4f}s')
-

pyg_lib/ops/__init__.py

@@ -352,9 +352,8 @@ def forward(
     @staticmethod
     def backward(ctx, out_grad: Tensor) -> Tuple[Union[Tensor, int]]:
         out, index, ptr = ctx.saved_tensors
-        in_grad = torch.ops.pyg.softmax_backward(
-            out, out_grad, index, ptr, ctx.num_nodes, ctx.dim
-        )
+        in_grad = torch.ops.pyg.softmax_backward(out, out_grad, index, ptr,
+                                                 ctx.num_nodes, ctx.dim)
 
         return in_grad, None, None, None, None