Apply suggestions from code review

Author: michaelosthege

pymc/distributions/continuous.py

@@ -337,7 +337,7 @@ def logcdf(value, lower, upper):
         )
 
     def get_moment(value, size, lower, upper):
-        lower = at.full(size, lower, dtype=aesara.config.floatX)
+        lower = at.full(size, lower, dtype=value.owner.op.inputs[2])
         upper = at.full(size, upper, dtype=aesara.config.floatX)
         return (lower + upper) / 2
 

pymc/distributions/discrete.py

@@ -396,7 +396,7 @@ def logcdf(value, p):
 
     def get_moment(value, size, p):
         p = at.full(size, p)
-        return at.switch(p < 0.5, np.int64(0), np.int64(1))
+        return at.switch(p < 0.5, at.zeros_like(value), at.ones_like(value))
 
     def _distr_parameters_for_repr(self):
         return ["p"]