Skip to content

Commit c2dea67

Browse files
committed
fix: sampling should use DistIter, but not collect it
1 parent 2992dec commit c2dea67

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

examples/random_clap.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -159,24 +159,24 @@ fn run_command_sample(count: Option<usize>, dist: DistributionAsCommand) -> Resu
159159
match dist {
160160
// multinomial should print `count` of Vec<uint>
161161
DistributionAsCommand::Multinomial { n, p } => {
162-
let samples = thread_rng().sample_iter(Multinomial::new(p, n)?);
162+
let sample_iter = thread_rng().sample_iter(Multinomial::new(p, n)?);
163163
print_multivariate_samples(
164164
count,
165-
samples.map(|v: na::DVector<u64>| {
165+
sample_iter.map(|v: na::DVector<u64>| {
166166
let vec: Vec<_> = v.into_iter().cloned().collect();
167167
vec
168168
}),
169169
)?;
170170
}
171171
// binomial should print `count` of uint
172172
DistributionAsCommand::Binomial { n, p } => {
173-
let samples: Vec<u64> = thread_rng().sample_iter(Binomial::new(p, n)?).collect();
174-
print_samples(count, samples)?;
173+
let sample_iter = thread_rng().sample_iter::<u64, Binomial>(Binomial::new(p, n)?);
174+
print_samples(count, sample_iter)?;
175175
}
176176
// normal should print `count` of float
177177
DistributionAsCommand::Normal { mu, sigma } => {
178-
let samples = thread_rng().sample_iter(Normal::new(mu, sigma)?);
179-
print_samples(count, samples)?
178+
let sample_iter = thread_rng().sample_iter(Normal::new(mu, sigma)?);
179+
print_samples(count, sample_iter)?
180180
}
181181
}
182182

0 commit comments

Comments
 (0)