Skip to content
Snippets Groups Projects
Commit 4e75c788 authored by Simon Ding's avatar Simon Ding
Browse files

revert

parent 73510d7e
Branches
Tags
1 merge request!26Observation slicing
Pipeline #109073 passed
...@@ -364,15 +364,13 @@ class Observation(BaseObservation): ...@@ -364,15 +364,13 @@ class Observation(BaseObservation):
obs = Observation.load(data_path, (lo, hi)) obs = Observation.load(data_path, (lo, hi))
obs.save(f"{target_folder}/{base_name}_{rank}.npz") obs.save(f"{target_folder}/{base_name}_{rank}.npz")
return "Done"
@staticmethod @staticmethod
def mpi_load(data_folder, base_name, full_data_set, nwork, comm=None): def mpi_load(data_folder, base_name, full_data_set, nwork, comm=None):
if master: if master:
from os.path import isdir from os.path import isdir
if not isdir(data_folder): if not isdir(data_folder):
end_message = Observation.split_data_file(full_data_set, comm.Get_size, data_folder, base_name, nwork) Observation.split_data_file(full_data_set, comm.Get_size, data_folder, base_name, nwork)
comm.bcast(end_message)
if comm is None: if comm is None:
return Observation.load(full_data_set) return Observation.load(full_data_set)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment