Fix rebase bugs

probabilistic-numerics · Aug 26, 2020 · bbbefcf · bbbefcf
1 parent b4cce51
commit bbbefcf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/probnum/filtsmooth/gaussfiltsmooth/gaussfiltsmooth.py b/src/probnum/filtsmooth/gaussfiltsmooth/gaussfiltsmooth.py
@@ -184,4 +184,4 @@ def linear_discrete_update(meanest, cpred, data, meascov, measmat, mpred):
     ccest = cpred @ measmat.T
     mean = mpred + ccest @ np.linalg.solve(covest, data - meanest)
     cov = cpred - ccest @ np.linalg.solve(covest.T, ccest.T)
-    return RandomVariable(distribution=Normal(mean, cov)), covest, ccest, meanest
+    return Normal(mean, cov), covest, ccest, meanest
diff --git a/tests/test_diffeq/test_odefiltsmooth/test_odefiltsmooth.py b/tests/test_diffeq/test_odefiltsmooth/test_odefiltsmooth.py
@@ -478,7 +478,7 @@ class TestPreconditioning(unittest.TestCase):
     """
 
     def setUp(self):
-        initdist = RandomVariable(distribution=Dirac(20 * np.ones(2)))
+        initdist = Dirac(20 * np.ones(2))
         self.ivp = ode.lotkavolterra([0.0, 1e-4], initdist)
         self.step = 1e-5
         self.prior = "ibm3"