Use compile_rv_inplace in SMC logp functions

pymc-devs · Jul 23, 2021 · b13da02 · b13da02
1 parent 2ff9011
commit b13da02
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pymc3/smc/smc.py b/pymc3/smc/smc.py
@@ -23,6 +23,7 @@
 from scipy.stats import multivariate_normal
 
 from pymc3.aesaraf import (
+    compile_rv_inplace,
     floatX,
     inputvars,
     join_nonshared_inputs,
@@ -264,8 +265,8 @@ def logp_forw(point, out_vars, vars, shared):
             "together with aesara.config.floatX == `float32`",
             UserWarning,
         )
-        f = aesara_function([inarray0], out_list[0], allow_input_downcast=True)
+        f = compile_rv_inplace([inarray0], out_list[0], allow_input_downcast=True)
     else:
-        f = aesara_function([inarray0], out_list[0])
+        f = compile_rv_inplace([inarray0], out_list[0])
         f.trust_input = True
     return f