diff --git a/nifty5/linearization.py b/nifty5/linearization.py index 30a9b57bad11b5e264a0997bde6c0e0e2bbefe10..43b184503e613e0e5192e71a08a1804776427cac 100644 --- a/nifty5/linearization.py +++ b/nifty5/linearization.py @@ -170,11 +170,11 @@ class Linearization(object): @staticmethod def make_partial_var(field, constants, want_metric=False): from .operators.scaling_operator import ScalingOperator - from .operators.simple_linear_operators import NullOperator + from .operators.block_diagonal_operator import BlockDiagonalOperator if len(constants) == 0: return Linearization.make_var(field, want_metric) else: ops = [ScalingOperator(0. if key in constants else 1., dom) for key, dom in field.domain.items()] - bdop = BlockDiagonalOperator(fielld.domain, tuple(ops)) + bdop = BlockDiagonalOperator(field.domain, tuple(ops)) return Linearization(field, bdop, want_metric=want_metric)