From e5ab07fb02aa42aa2c187d0ff94da305ee957b09 Mon Sep 17 00:00:00 2001 From: Philipp Arras <c@philipp-arras.de> Date: Tue, 4 Feb 2025 12:35:08 +0100 Subject: [PATCH] Linearization: check that "[]" works only on MultiDomains --- src/linearization.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/linearization.py b/src/linearization.py index 32e3996a7..5f04a018d 100644 --- a/src/linearization.py +++ b/src/linearization.py @@ -17,6 +17,7 @@ import numpy as np +from .multi_domain import MultiDomain from .operators.operator import Operator from .sugar import makeOp from .utilities import check_object_identity @@ -124,6 +125,8 @@ class Linearization(Operator): return self._metric def __getitem__(self, name): + if not isinstance(self.domain, MultiDomain): + return NotImplemented return self.new(self._val[name], self._jac.ducktape_left(name)) def __neg__(self): -- GitLab