Skip to content

Commit db5556c

Browse files
update sample cpu doc
1 parent c0e062e commit db5556c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/queries/sample_cpu.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ function sample(pc::ProbCircuit, num_samples, types; rng = default_rng(), Float=
1515
end
1616

1717
"""
18-
sample(pc::ProbCircuit, num_samples; rng = default_rng())
18+
sample(pc::ProbCircuit, num_samples; batch_size, rng = default_rng())
1919
2020
Generate `num_samples` from the joint distribution of the circuit conditioned on the `data`.
2121
"""
@@ -73,4 +73,4 @@ function sample_rec!(node::ProbCircuit, states, values, data; s_id, ex_id, rng,
7373
sample_rec!(sampled_child, states, values, data; s_id, ex_id, rng, node2idx)
7474
end
7575
return nothing
76-
end
76+
end

0 commit comments

Comments
 (0)