Commit 58180567 authored by Simeon Doetsch's avatar Simeon Doetsch

Fixed typo in Simulation.reduce()

parent 77f55c12
......@@ -195,7 +195,7 @@ class Simulation:
if dtype is None:
dtype = first.dtype
if first.shape == () or first.shape == (1,):
return np.fromiter((func(d) for d in self), dtype=dtype, count=N)
return np.fromiter((func(d) for d in self), dtype=dtype, count=len(self))
else:
shape = (len(self), *first.shape)
res = np.empty(shape, dtype=dtype)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment