debug

src/__init__.py

@@ -90,7 +90,7 @@ from .library.adjust_variances import (make_adjust_variances_hamiltonian,
 from .library.gridder import Gridder
 from .library.correlated_fields import CorrelatedFieldMaker
 from .library.correlated_fields_simple import SimpleCorrelatedField
-from .library.meanfield_model import MeanfieldModel
+from .library.variational_models import MeanfieldModel
 
 from . import extra
 

src/minimization/stochastic_minimizer.py

@@ -15,7 +15,7 @@ class ADVIOptimizer(Minimizer):
 
     def _step(self, position, gradient):
         self.s = self.alpha * gradient**2 + (1-self.alpha)*self.s
-        self.rho = self.eta * self.counter**(-0.5+ self.epsilon) / (self.tau + sqrt(self.s))
+        self.rho = self.eta * self.counter**(-0.5+ self.epsilon) / (self.tau + (self.s).sqrt())
         new_position = position - self.rho * gradient
         self.counter += 1
         return new_position