Fix Gaussian curvature

38ed0ec7 · Philipp Arras · 73c7b018 · 38ed0ec7
Commit 38ed0ec7 authored Jun 29, 2018 by Philipp Arras
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

nifty5/library/gaussian_energy.py nifty5/library/gaussian_energy.py +3 -1

No files found.
--- a/nifty5/library/gaussian_energy.py
+++ b/nifty5/library/gaussian_energy.py
@@ -61,4 +61,6 @@ class GaussianEnergy(Energy):
    @property
    @memo
    def curvature(self):
-        return SandwichOperator.make(self._inp.gradient)
+        if self._cov is None:
+            return SandwichOperator.make(self._inp.gradient)
+        return SandwichOperator.make(self._inp.gradient, self._cov.inverse)