We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 02bb110 commit f13f9e4Copy full SHA for f13f9e4
pymc_bart/bart.py
@@ -22,7 +22,7 @@
22
import numpy.typing as npt
23
import pytensor.tensor as pt
24
from pandas import DataFrame, Series
25
-from pymc.distributions.distribution import Distribution, _moment
+from pymc.distributions.distribution import Distribution, _support_point
26
from pymc.logprob.abstract import _logprob
27
from pytensor.tensor.random.op import RandomVariable
28
@@ -164,7 +164,7 @@ def __new__(
164
165
Distribution.register(BARTRV)
166
167
- @_moment.register(BARTRV)
+ @_support_point.register(BARTRV)
168
def get_moment(rv, size, *rv_inputs):
169
return cls.get_moment(rv, size, *rv_inputs)
170
0 commit comments