Skip to content

Commit 50697be

Browse files
committed
Use add instead of sum in logp to avoid casting everything explicitly to the same dtype
1 parent fd11cf0 commit 50697be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc/model/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -748,7 +748,7 @@ def logp(
748748
if not sum:
749749
return logp_factors
750750

751-
logp_scalar = pt.sum([pt.sum(factor) for factor in logp_factors])
751+
logp_scalar = pt.add(*[pt.sum(factor) for factor in logp_factors])
752752
logp_scalar_name = "__logp" if jacobian else "__logp_nojac"
753753
if self.name:
754754
logp_scalar_name = f"{logp_scalar_name}_{self.name}"

0 commit comments

Comments
 (0)