diff --git a/nifty5/operators/energy_operators.py b/nifty5/operators/energy_operators.py index ef3fe76333e2266f7e2b1d1e1266abf524f5d046..91a23db8fbe5710ff2df8b78aaaea33ef54bd8ea 100644 --- a/nifty5/operators/energy_operators.py +++ b/nifty5/operators/energy_operators.py @@ -20,6 +20,7 @@ import numpy as np from .. import utilities from ..domain_tuple import DomainTuple from ..field import Field +from ..multi_field import MultiField from ..linearization import Linearization from ..sugar import makeDomain, makeOp from .linear_operator import LinearOperator @@ -122,7 +123,8 @@ class GaussianEnergy(EnergyOperator): def __init__(self, mean=None, covariance=None, domain=None): if mean is not None and not isinstance(mean, Field): - raise TypeError + if not isinstance(mean, MultiField): + raise TypeError if covariance is not None and not isinstance(covariance, LinearOperator): raise TypeError