From b8344c6049ec3f900b28a1c82f575efdad263f13 Mon Sep 17 00:00:00 2001 From: Philipp Arras <parras@mpa-garching.mpg.de> Date: Fri, 16 Jul 2021 12:38:19 +0200 Subject: [PATCH] Simplify --- src/utilities.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/src/utilities.py b/src/utilities.py index 28dc5a6ba..8fe6f9901 100644 --- a/src/utilities.py +++ b/src/utilities.py @@ -18,6 +18,7 @@ import collections from functools import reduce from itertools import product +import pickle import numpy as np @@ -451,12 +452,7 @@ def check_MPI_equality(obj, comm): def _MPI_unique(obj, comm): - from collections.abc import Hashable - import pickle - objects = comm.allgather(obj) - if not isinstance(objects[0], Hashable) or isinstance(objects[0], np.random.SeedSequence): - objects = [pickle.dumps(oo) for oo in objects] - return len(set(objects)) == 1 + return len(set(comm.allgather(pickle.dumps(obj)))) == 1 def check_MPI_synced_random_state(comm): -- GitLab