fixes

nifty5/library/correlated_fields.py

@@ -46,12 +46,17 @@ class CorrelatedField(Operator):
         self._p_space = amplitude_model.target[0]
         self._power_distributor = PowerDistributor(self._h_space,
                                                    self._p_space)
+        self._domain = MultiDomain.union(
+            (self._amplitude_model.domain,
+             MultiDomain.make({"xi": self._h_space})))
 
     @property
     def domain(self):
-        return MultiDomain.union(
-            (self._amplitude_model.domain,
-             MultiDomain.make({"xi": self._h_space})))
+        return self._domain
+
+    @property
+    def target(self):
+        return self._ht.target
 
     def __call__(self, x):
         A = self._power_distributor(self._amplitude_model(x))

test/test_energies/test_consistency.py

@@ -40,7 +40,7 @@ class Energy_Tests(unittest.TestCase):
     def testGaussian(self, space, seed):
         model = self.make_model(
             space_key='s1', space=space, seed=seed)['s1']
-        energy = ift.GaussianEnergy()
+        energy = ift.GaussianEnergy(domain=space)
         ift.extra.check_value_gradient_consistency(energy, model)
 
 #     @expand(product(
@@ -81,7 +81,7 @@ class Energy_Tests(unittest.TestCase):
         model = self.make_model(
             space_key='s1', space=space, seed=seed)['s1']
         model = model.exp()
-        lh = ift.GaussianEnergy()
+        lh = ift.GaussianEnergy(domain=space)
         hamiltonian = ift.Hamiltonian(lh)
         ift.extra.check_value_gradient_consistency(hamiltonian, model)
         S = ift.ScalingOperator(1., space)