Skip to content

Commit 116ae6b

Browse files
Apply suggestions from code review
1 parent beb5bb9 commit 116ae6b

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pymc/distributions/continuous.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,7 @@ def logcdf(value, lower, upper):
337337
)
338338

339339
def get_moment(value, size, lower, upper):
340-
lower = at.full(size, lower, dtype=aesara.config.floatX)
340+
lower = at.full(size, lower, dtype=value.owner.op.inputs[2])
341341
upper = at.full(size, upper, dtype=aesara.config.floatX)
342342
return (lower + upper) / 2
343343

pymc/distributions/discrete.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -396,7 +396,7 @@ def logcdf(value, p):
396396

397397
def get_moment(value, size, p):
398398
p = at.full(size, p)
399-
return at.switch(p < 0.5, np.int64(0), np.int64(1))
399+
return at.switch(p < 0.5, at.zeros_like(value), at.ones_like(value))
400400

401401
def _distr_parameters_for_repr(self):
402402
return ["p"]

0 commit comments

Comments
 (0)