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

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

torch_geometric/datasets/nell.py

@@ -52,8 +52,12 @@ class NELL(InMemoryDataset):
 
     url = 'http://www.cs.cmu.edu/~zhiliny/data/nell_data.tar.gz'
 
-    def __init__(self, root: str, transform: Optional[Callable] = None,
-                 pre_transform: Optional[Callable] = None, ):
+    def __init__(
+        self,
+        root: str,
+        transform: Optional[Callable] = None,
+        pre_transform: Optional[Callable] = None,
+    ):
         super().__init__(root, transform, pre_transform)
         self.data, self.slices = torch.load(self.processed_paths[0])
 

torch_geometric/datasets/ppi.py

@@ -60,10 +60,14 @@ class PPI(InMemoryDataset):
 
     url = 'https://data.dgl.ai/dataset/ppi.zip'
 
-    def __init__(self, root: str, split: str = 'train',
-                 transform: Optional[Callable] = None,
-                 pre_transform: Optional[Callable] = None,
-                 pre_filter: Optional[Callable] = None, ):
+    def __init__(
+        self,
+        root: str,
+        split: str = 'train',
+        transform: Optional[Callable] = None,
+        pre_transform: Optional[Callable] = None,
+        pre_filter: Optional[Callable] = None,
+    ):
 
         assert split in ['train', 'val', 'test']
 

torch_geometric/datasets/qm7.py

@@ -44,9 +44,13 @@ class QM7b(InMemoryDataset):
 
     url = 'https://deepchemdata.s3-us-west-1.amazonaws.com/datasets/qm7b.mat'
 
-    def __init__(self, root: str, transform: Optional[Callable] = None,
-                 pre_transform: Optional[Callable] = None,
-                 pre_filter: Optional[Callable] = None, ):
+    def __init__(
+        self,
+        root: str,
+        transform: Optional[Callable] = None,
+        pre_transform: Optional[Callable] = None,
+        pre_filter: Optional[Callable] = None,
+    ):
         super().__init__(root, transform, pre_transform, pre_filter)
         self.data, self.slices = torch.load(self.processed_paths[0])
 

torch_geometric/datasets/zinc.py

@@ -86,9 +86,15 @@ class ZINC(InMemoryDataset):
     split_url = ('https://raw.githubusercontent.com/graphdeeplearning/'
                  'benchmarking-gnns/master/data/molecules/{}.index')
 
-    def __init__(self, root: str, subset: bool = False, split: str = 'train',
-                 transform: Optional[Callable] = None, pre_transform: Optional[Callable] =None,
-                 pre_filter: Optional[Callable] =None, ):
+    def __init__(
+        self,
+        root: str,
+        subset: bool = False,
+        split: str = 'train',
+        transform: Optional[Callable] = None,
+        pre_transform: Optional[Callable] = None,
+        pre_filter: Optional[Callable] = None,
+    ):
         self.subset = subset
         assert split in ['train', 'val', 'test']
         super().__init__(root, transform, pre_transform, pre_filter)