fix density setting (#35)

mathematicalmichael · Feb 8, 2021 · f8d3f9c · f8d3f9c
1 parent c3a13c1
commit f8d3f9c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/mud/base.py b/src/mud/base.py
@@ -44,7 +44,8 @@ def set_initial(self, distribution=None):
                 mn = np.min(self.domain, axis=1)
                 mx = np.max(self.domain, axis=1)
                 distribution = dist.uniform(loc=mn, scale=mx - mn)
-            distribution = dist.norm()
+            else:
+                distribution = dist.norm()
         initial_dist = distribution
         self._in = initial_dist.pdf(self.X).prod(axis=1)
         self._up = None
@@ -122,7 +123,8 @@ def set_prior(self, distribution=None):
                 mn = np.min(self.domain, axis=1)
                 mx = np.max(self.domain, axis=1)
                 distribution = dist.uniform(loc=mn, scale=mx - mn)
-            distribution = dist.norm()
+            else:
+                distribution = dist.norm()
         prior_dist = distribution
         self._pr = prior_dist.pdf(self.X).prod(axis=1)
         self._ps = None