lorenzo.volpi
/
QuAcc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refatored create_dataset_dir

This commit is contained in:
Lorenzo Volpi 2023-11-26 16:33:43 +01:00
parent e05dfd4a16
commit 8096c0842e
1 changed files with 4 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
quacc/utils.py
View File

@ -41,24 +41,14 @@ def fmt_line_md(s):
return f"> {s} \n" return f"> {s} \n"
def create_dataser_dir(dir_name, update=False, create_md=False): def create_dataser_dir(dir_name, update=False):
base_out_dir = Path(env.OUT_DIR_NAME) dataset_dir = Path(env.OUT_DIR_NAME) / dir_name
os.makedirs(base_out_dir, exist_ok=True)
dataset_dir = base_out_dir / dir_name
env.OUT_DIR = dataset_dir env.OUT_DIR = dataset_dir
if update: if update:
if not dataset_dir.exists(): os.makedirs(dataset_dir, exist_ok=True)
os.mkdir(dataset_dir)
else: else:
shutil.rmtree(dataset_dir, ignore_errors=True) shutil.rmtree(dataset_dir, ignore_errors=True)
os.mkdir(dataset_dir) os.makedirs(dataset_dir)
if create_md:
plot_dir_path = dataset_dir / "plot"
env.PLOT_OUT_DIR = plot_dir_path
if not plot_dir_path.exists():
os.mkdir(plot_dir_path)
def get_quacc_home(): def get_quacc_home():