File size: 1,083 Bytes
31d1d63 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | import argparse
import os
from pathlib import Path
import subprocess
import sys
ROOT = Path(__file__).resolve().parent
ap = argparse.ArgumentParser(description="Plot the saved paper results")
ap.add_argument(
"--only", choices=["all", "mb", "aib9_robustness", "p53"], default="all"
)
args = ap.parse_args()
commands = {
"mb": [
["make_mb_panels.py", "--from-cache"],
["make_mb_unconditional.py", "--from-cache"],
],
"aib9_robustness": [["replot_aib9_robustness.py"]],
"p53": [
["paper_figures/plot/p53_discovery_plot.py"],
["paper_figures/plot/p53_preference_plot.py"],
],
}
env = os.environ.copy()
for name in ["OMP_NUM_THREADS", "OPENBLAS_NUM_THREADS", "MKL_NUM_THREADS"]:
env[name] = "1"
env["MPLBACKEND"] = "Agg"
env["MPLCONFIGDIR"] = str(ROOT / ".mplconfig")
for group, jobs in commands.items():
if args.only not in ["all", group]:
continue
for command in jobs:
print("Running:", " ".join(command), flush=True)
subprocess.run([sys.executable, *command], cwd=ROOT, env=env, check=True)
|