We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0226047 commit 9ab831dCopy full SHA for 9ab831d
pymc3/aesaraf.py
@@ -156,7 +156,7 @@ def change_rv_size(
156
size = rv_node.op._infer_shape(size, dist_params)
157
new_size = tuple(np.atleast_1d(new_size)) + tuple(size)
158
159
- # Make sure the new size is int64 so that it doesn't unnecessarily pick
+ # Make sure the new size is a tensor. This helps to not unnecessarily pick
160
# up a `Cast` in some cases
161
new_size = at.as_tensor(new_size, ndim=1, dtype="int64")
162
0 commit comments