From f8d3f9c65dbb29668e20d07f0433198b3011c7a4 Mon Sep 17 00:00:00 2001
From: Michael Pilosov <40366263+mathematicalmichael@users.noreply.github.com>
Date: Mon, 8 Feb 2021 01:06:46 -0700
Subject: [PATCH] fix density setting (#35)

---
 src/mud/base.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/mud/base.py b/src/mud/base.py
index 622ebe3..37f6546 100644
--- 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