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