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

for more information, see https://pre-commit.ci
pyg-team · Aug 28, 2023 · 928ca79 · 928ca79
1 parent 298f3d4
commit 928ca79
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torch_geometric/loader/cache.py b/torch_geometric/loader/cache.py
@@ -28,9 +28,9 @@ def __init__(self, loader: DataLoader, device: torch.device):
 
         # register default hooks
         self.register_attr_hooks([
-            lambda b: b.n_id,
-            lambda b: b.adj_t if hasattr(b, 'adj_t') else b.edge_index,
-            lambda b: b.batch_size])
+            lambda b: b.n_id, lambda b: b.adj_t
+            if hasattr(b, 'adj_t') else b.edge_index, lambda b: b.batch_size
+        ])
 
     def _check_if_reg_is_open(self) -> None:
         if not self.hook_reg_open: