Fixed deprecation bug in SPM.score().

QInfer · Aug 6, 2018 · 0c1aeb0 · 0c1aeb0
1 parent 2376df4
commit 0c1aeb0
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/qinfer/test_models.py b/src/qinfer/test_models.py
@@ -188,8 +188,13 @@ def score(self, outcomes, modelparams, expparams, return_L=False):
         new_eps['w_'] = 0
         new_eps['t'] = expparams
 
-        return super(SimplePrecessionModel, self).score(outcomes, modelparams, new_eps, return_L)
-
+        q = super(SimplePrecessionModel, self).score(outcomes, modelparams, new_eps, return_L=False)
+
+        if return_L:
+            return q, self.likelihood(outcomes, modelparams, expparams)
+        else:
+            return q
+
 class CoinModel(FiniteOutcomeModel, DifferentiableModel):
     r"""
     Arguably the simplest possible model; the unknown model parameter