diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 59a9ee9..6249b16 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -15,12 +15,12 @@ repos: args: ["--autofix", "--no-ensure-ascii"] - repo: https://github.com/asottile/pyupgrade - rev: v3.21.1 + rev: v3.21.2 hooks: - id: pyupgrade - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.14.5 + rev: v0.15.8 hooks: - id: ruff args: ["--fix"] diff --git a/src/finch/einstein.py b/src/finch/einstein.py index 34281cd..a5c4de3 100644 --- a/src/finch/einstein.py +++ b/src/finch/einstein.py @@ -196,7 +196,7 @@ def run(self, xp, kwargs): arg = self.arg.run(xp, loops, kwargs) axis = tuple(i for i in range(len(loops)) if loops[i] not in self.idxs) if self.op is not None: - op = getattr(xp, reduction_ops.get(self.op, None)) + op = getattr(xp, reduction_ops.get(self.op)) val = op(arg, axis=axis) else: assert set(self.idxs) == set(loops)