Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools.git synced 2024-12-22 15:18:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Make sure dumpfolder created only on rank 0

Browse source
This commit is contained in:
Richard Stiskalek 2024-02-27 13:28:41 +00:00
parent 66d9f46919
commit fb93f85543
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
scripts/field_los.py
View file

@ -19,7 +19,7 @@ from argparse import ArgumentParser
from datetime import datetime
from gc import collect
from os import makedirs, remove, rmdir
from os.path import exists, isdir, join
from os.path import exists, join
import csiborgtools
import numpy as np
@ -229,9 +229,6 @@ if __name__ == "__main__":
parser.add_argument("--grid", type=int, help="Grid resolution.")
args = parser.parse_args()
out_folder = "/mnt/extraspace/rstiskalek/csiborg_postprocessing/field_los"
dump_folder = join(out_folder,
f"temp_{str(datetime.now())}".replace(" ", "_"))
rmax = 200
dr = 0.1
smooth_scales = None
@ -240,10 +237,16 @@ if __name__ == "__main__":
paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring)
nsims = get_nsims(args, paths)
out_folder = "/mnt/extraspace/rstiskalek/csiborg_postprocessing/field_los"
# Create the dumping folder.
if comm.Get_rank() == 0 and not isdir(dump_folder):
if comm.Get_rank() == 0:
dump_folder = join(out_folder,
f"temp_{str(datetime.now())}".replace(" ", "_"))
print(f"Creating folder `{dump_folder}`.")
makedirs(dump_folder)
else:
dump_folder = None
dump_folder = comm.bcast(dump_folder, root=0)
# Get the line of sight RA/dec coordinates.
pos = get_los(args.catalogue, comm)