Skip to content
Snippets Groups Projects
Commit e5ab07fb authored by Philipp Arras's avatar Philipp Arras
Browse files

Linearization: check that "[]" works only on MultiDomains

parent 8608678a
No related branches found
No related tags found
1 merge request!994Two small fixups
Pipeline #237224 passed
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
import numpy as np import numpy as np
from .multi_domain import MultiDomain
from .operators.operator import Operator from .operators.operator import Operator
from .sugar import makeOp from .sugar import makeOp
from .utilities import check_object_identity from .utilities import check_object_identity
...@@ -124,6 +125,8 @@ class Linearization(Operator): ...@@ -124,6 +125,8 @@ class Linearization(Operator):
return self._metric return self._metric
def __getitem__(self, name): def __getitem__(self, name):
if not isinstance(self.domain, MultiDomain):
return NotImplemented
return self.new(self._val[name], self._jac.ducktape_left(name)) return self.new(self._val[name], self._jac.ducktape_left(name))
def __neg__(self): def __neg__(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment