diff --git a/resolve/data/observation.py b/resolve/data/observation.py index e1278e02b99cae9eef8661dcdb7a520bc809fa46..88311036e1736d6050eadecd09e6c6aff541485e 100644 --- a/resolve/data/observation.py +++ b/resolve/data/observation.py @@ -364,16 +364,19 @@ class Observation(BaseObservation): obs = Observation.load(data_path, (lo, hi)) obs.save(f"{target_folder}/{base_name}_{rank}.npz") + return "Done" + @staticmethod def mpi_load(data_folder, base_name, full_data_set, nwork, comm=None): if master: from os.path import isdir if not isdir(data_folder): - Observation.split_data_file(full_data_set, comm.Get_size, data_folder, base_name, nwork) + end_message = Observation.split_data_file(full_data_set, comm.Get_size, data_folder, base_name, nwork) + comm.bcast(end_message) if comm is None: return Observation.load(full_data_set) - comm.barrier() + comm.Barrier() return Observation.load(f"{data_folder}/{base_name}_{comm.Get_rank()}.npz")