Skip to content

Commit f4a251b

Browse files
committed
minor updates
1 parent 4eafe67 commit f4a251b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lectures/multi_hyper.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def moments(urn, n):
216216
217217
218218
def simulate(urn, n, size=1, seed=0):
219-
"""Simulate multivariate hypergeometric draws efficiently."""
219+
"""Simulate multivariate hypergeometric draws."""
220220
K_arr, c = urn.K, urn.c
221221
n = int(n)
222222
size = int(size)
@@ -226,7 +226,7 @@ def simulate(urn, n, size=1, seed=0):
226226
return jnp.zeros((0, c), dtype=jnp.int32)
227227
228228
def draw_once(key):
229-
"""Generate one draw via a lax.scan loop (JIT friendly)."""
229+
"""Generate one draw via a lax.scan loop."""
230230
counts0 = jnp.zeros(c, dtype=jnp.int32)
231231
232232
def body_fn(carry, _):

0 commit comments

Comments
 (0)