Commit ac7ee7f8 authored by Martin Reinecke's avatar Martin Reinecke
Browse files

extra coolness

parent 3021c617
Pipeline #75660 passed with stages
in 13 minutes and 41 seconds
......@@ -264,9 +264,7 @@ class MetricGaussianKL(Energy):
rank_lo_hi = [_shareRange(self._n_samples, ntask, i) for i in range(ntask)]
lo, hi = rank_lo_hi[rank]
vals = [None]*lo + list(obj) + [None]*(self._n_samples-hi)
who = np.zeros(len(vals), dtype=np.int32)
for t, (l,h) in enumerate(rank_lo_hi):
who[l:h] = t
who = [t for t, (l, h) in enumerate(rank_lo_hi) for cnt in range(h-l)]
step = 1
while step < self._n_samples:
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