inspect_dag_group_run.py 469 Bytes
Newer Older
BO ZHANG's avatar
BO ZHANG committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import sys
import os

dag_group_run_hash = sys.argv[1]

assert len(dag_group_run_hash) == 40
import joblib

file_path = os.path.join(
    os.getenv("HOME"),
    "csst_dag",
    f"{dag_group_run_hash}.joblib",
)

data = joblib.load(file_path)
dag_group_run = data["dag_group_run"]
dag_run_list = data["dag_run_list"]

# print dag_group_run and dag_run_list
print("`dag_group_run`:")
print(f"  - {dag_group_run}")
for dag_run in dag_run_list:
    print(f"  - {dag_run}")