Commit 7d0d885f authored by Martin Reinecke's avatar Martin Reinecke Committed by Philipp Arras
Browse files


parent 94d27fd0
Pipeline #105700 passed with stages
in 20 minutes and 52 seconds
......@@ -71,7 +71,7 @@ class JaxOperator(Operator):
res = _jax2np(self._func(x.val))
if isinstance(res, dict):
if not isinstance(self._target, MultiDomain):
raise TypeError(("Jax function return a dictionary although the "
raise TypeError(("Jax function returns a dictionary although the "
"target of the operator is a DomainTuple."))
if set(res.keys()) != set(self._target.keys()):
raise ValueError(("Keys do not match:\n"
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment