simplify

Author: aloctavodia

pymc/bart/bart.py

@@ -20,7 +20,6 @@
 from pandas import DataFrame, Series
 
 from pymc.distributions.distribution import NoDistribution, _get_moment
-from pymc.distributions.shape_utils import rv_size_is_none
 
 __all__ = ["BART"]
 
@@ -139,9 +138,7 @@ def logp(x, *inputs):
 
     @classmethod
     def get_moment(cls, rv, size, *rv_inputs):
-        mean = rv.Y.mean()
-        if not rv_size_is_none(size):
-            mean = at.fill(size, mean)
+        mean = at.fill(size, rv.Y.mean())
         return mean