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

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

torch_geometric/sampler/base.py

@@ -1,10 +1,12 @@
 from abc import ABC, abstractmethod
-from typing import Dict, NamedTuple, Tuple, Union, List
+from typing import Dict, List, NamedTuple, Tuple, Union
+
 import torch
-from torch_geometric.typing import NodeType, EdgeType
+
 from torch_geometric.data import Data, HeteroData
 from torch_geometric.data.feature_store import FeatureStore
 from torch_geometric.data.graph_store import GraphStore
+from torch_geometric.typing import EdgeType, NodeType
 
 # An input to a sampler is either a list or tensor of node indices:
 SamplerInput = Union[List[int], torch.Tensor]

torch_geometric/sampler/neighbor_sampler.py

@@ -11,7 +11,11 @@
     to_csc,
     to_hetero_csc,
 )
-from torch_geometric.sampler.base import BaseSampler, SamplerInput, SamplerOutput
+from torch_geometric.sampler.base import (
+    BaseSampler,
+    SamplerInput,
+    SamplerOutput,
+)
 from torch_geometric.typing import NumNeighbors
 
 
@@ -205,4 +209,4 @@ def __call__(self, index: SamplerInput) -> SamplerOutput:
         elif self.data_cls == 'custom' or issubclass(self.data_cls,
                                                      HeteroData):
             return self._hetero_sparse_neighbor_sample(
-                {self.input_type: index}) + (index.numel(), )
+                {self.input_type: index}) + (index.numel(), )