summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorStan Seibert <stan@mtrr.org>2011-09-08 11:36:52 -0400
committerStan Seibert <stan@mtrr.org>2011-09-08 11:36:52 -0400
commit7cae8c3b343d1bde0320ff2ea224cc1ef79889df (patch)
tree5e9ea0bca12bb24f49de6a6b53e7bb16d9793044
parentaeb16cf530767cd16c23c258802325b726e577d7 (diff)
downloadchroma-7cae8c3b343d1bde0320ff2ea224cc1ef79889df.tar.gz
chroma-7cae8c3b343d1bde0320ff2ea224cc1ef79889df.tar.bz2
chroma-7cae8c3b343d1bde0320ff2ea224cc1ef79889df.zip
repeating_iterator needs a number of reps
-rwxr-xr-xsim.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/sim.py b/sim.py
index 931858f..26835f9 100755
--- a/sim.py
+++ b/sim.py
@@ -110,7 +110,7 @@ class Simulation(object):
self.gpu_pdf.clear_pdf()
if nreps > 1:
- iterable = repeating_iterator(iterable)
+ iterable = repeating_iterator(iterable, nreps)
for ev in iterable:
gpu_photons = gpu.GPUPhotons(ev.photons_beg)
@@ -139,7 +139,7 @@ class Simulation(object):
iterable = self.photon_generator.generate_events(iterable)
if nreps > 1:
- iterable = repeating_iterator(iterable)
+ iterable = repeating_iterator(iterable, nreps)
for ev in iterable:
gpu_photons = gpu.GPUPhotons(ev.photons_beg)