Skip to content

Commit 1716082

Browse files
Edit misleading comment to be more accurate
1 parent c528fde commit 1716082

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: pymc3/aesaraf.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def change_rv_size(
156156
size = rv_node.op._infer_shape(size, dist_params)
157157
new_size = tuple(np.atleast_1d(new_size)) + tuple(size)
158158

159-
# Make sure the new size is int64 so that it doesn't unnecessarily pick
159+
# Make sure the new size is a tensor. This helps to not unnecessarily pick
160160
# up a `Cast` in some cases
161161
new_size = at.as_tensor(new_size, ndim=1, dtype="int64")
162162

0 commit comments

Comments
 (0)