From c389e60636a340ffa357a998129b8591308dbab1 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 25 Aug 2023 13:35:12 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- torch_geometric/loader/cache.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torch_geometric/loader/cache.py b/torch_geometric/loader/cache.py index 581d2882747b..e9e5b91a55a8 100644 --- 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: