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

The property now returns all samples

parent 7e1cd862
......@@ -78,8 +78,12 @@ class KL_Energy(Energy):
def metric(self):
if ntask > 1:
raise ValueError("not supported when MPI is active")
return self._metric
def samples(self):
return self._samples
res = _comm.allgather(self._samples)
res = [item for sublist in res for item in sublist]
return res
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