From d304bf8c352585f933e94e658ea33865b423c83a Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Thu, 7 Nov 2019 14:13:33 +0100
Subject: [PATCH] Simplification

---
 nifty5/library/correlated_fields.py | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/nifty5/library/correlated_fields.py b/nifty5/library/correlated_fields.py
index 1c914ac19..e3b04e93b 100644
--- a/nifty5/library/correlated_fields.py
+++ b/nifty5/library/correlated_fields.py
@@ -205,14 +205,9 @@ class _Amplitude(Operator):
         mask = np.zeros(target.shape)
         mask[0] = vol
         adder = Adder(from_global_data(target, mask))
-        self._op = adder @ ((expander @ fluctuations)*normal_ampl)
-
-        self._domain = self._op.domain
-        self._target = self._op.target
-
-    def apply(self, x):
-        self._check_input(x)
-        return self._op(x)
+        op = adder @ ((expander @ fluctuations)*normal_ampl)
+        self.apply = op.apply
+        self._domain, self._target = op.domain, op.target
 
 
 class CorrelatedFieldMaker:
-- 
GitLab