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

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

Author: pre-commit-ci[bot]

torch_geometric/nn/models/gpse.py

@@ -9,6 +9,7 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+from torch.nn import Module
 from tqdm import trange
 
 import torch_geometric.transforms as T
@@ -20,7 +21,6 @@
     global_max_pool,
     global_mean_pool,
 )
-from torch.nn import Module
 from torch_geometric.nn.resolver import activation_resolver
 from torch_geometric.utils import to_dense_batch
 
@@ -716,8 +716,9 @@ def forward(self, x, pos_enc):
 
 
 @torch.no_grad()
-def gpse_process(model: Module, data: Data, rand_type: str, use_vn: bool = True,
-                 bernoulli_thresh: float = 0.5, neighbor_loader: bool = False,
+def gpse_process(model: Module, data: Data, rand_type: str,
+                 use_vn: bool = True, bernoulli_thresh: float = 0.5,
+                 neighbor_loader: bool = False,
                  num_neighbors: List[int] = [30, 20, 10], fillval: int = 5,
                  layers_mp: int = None, **kwargs) -> torch.Tensor:
     r"""Processes the data using the :class:`GPSE` model to generate and append