Update pymc/distributions/timeseries.py

canyon289 · ricardoV94 · canyon289 · commit 92aae53fbdd8 · 2022-04-03T11:37:16.000-07:00
Co-authored-by: Ricardo Vieira &lt;28983449+ricardoV94@users.noreply.github.com&gt;
diff --git a/pymc/distributions/timeseries.py b/pymc/distributions/timeseries.py
@@ -112,7 +112,7 @@ def rng_fn(
             )
             dist_shape = (*bcast_shape, int(steps))
 
-        # If size is None then the returned series should be (size, 1+steps)
+        # If size is None then the returned series should be (*size, 1+steps)
         else:
             init_size = (*size, 1)
             dist_shape = (*size, int(steps))

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def rng_fn(`
`112`	`112`	`)`
`113`	`113`	`dist_shape = (*bcast_shape, int(steps))`
`114`	`114`
`115`		`- # If size is None then the returned series should be (size, 1+steps)`
	`115`	`+ # If size is None then the returned series should be (*size, 1+steps)`
`116`	`116`	`else:`
`117`	`117`	`init_size = (*size, 1)`
`118`	`118`	`dist_shape = (*size, int(steps))`