From bbbefcfe44d2089464919a570be437e2a4080ada Mon Sep 17 00:00:00 2001 From: Marvin Pfoertner Date: Wed, 26 Aug 2020 18:21:22 +0200 Subject: [PATCH] Fix rebase bugs --- src/probnum/filtsmooth/gaussfiltsmooth/gaussfiltsmooth.py | 2 +- tests/test_diffeq/test_odefiltsmooth/test_odefiltsmooth.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/probnum/filtsmooth/gaussfiltsmooth/gaussfiltsmooth.py b/src/probnum/filtsmooth/gaussfiltsmooth/gaussfiltsmooth.py index 8434916bc5..2df32edfc2 100644 --- 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 index 79dab7d297..d03685a203 100644 --- 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"