rename moment to support_point (#154)

pymc-devs · Mar 15, 2024 · f13f9e4 · f13f9e4
1 parent 02bb110
commit f13f9e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymc_bart/bart.py b/pymc_bart/bart.py
@@ -22,7 +22,7 @@
 import numpy.typing as npt
 import pytensor.tensor as pt
 from pandas import DataFrame, Series
-from pymc.distributions.distribution import Distribution, _moment
+from pymc.distributions.distribution import Distribution, _support_point
 from pymc.logprob.abstract import _logprob
 from pytensor.tensor.random.op import RandomVariable
 
@@ -164,7 +164,7 @@ def __new__(
 
         Distribution.register(BARTRV)
 
-        @_moment.register(BARTRV)
+        @_support_point.register(BARTRV)
         def get_moment(rv, size, *rv_inputs):
             return cls.get_moment(rv, size, *rv_inputs)