Commit 07625e4c authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent 5511f04a
...@@ -7,7 +7,7 @@ from astropy import table ...@@ -7,7 +7,7 @@ from astropy import table
from .._dfs import DFS, dfs from .._dfs import DFS, dfs
from ..hash import generate_sha1_from_time from ..hash import generate_sha1_from_time
from ._dispatcher import Dispatcher from ._dispatcher import Dispatcher, override_common_keys
DAG_CONFIG_DIR = os.path.join( DAG_CONFIG_DIR = os.path.join(
...@@ -108,10 +108,10 @@ class BaseDAG: ...@@ -108,10 +108,10 @@ class BaseDAG:
) )
task_list = self.dispatcher(filtered_plan_basis, filtered_data_basis) task_list = self.dispatcher(filtered_plan_basis, filtered_data_basis)
dag_run_list = [] dag_run_list = []
for task in task_list: for this_task in task_list:
dag_run = self.gen_dag_run( dag_run = self.gen_dag_run(
dag_group_run=dag_group_run, dag_group_run=dag_group_run,
**task, **this_task["task"],
) )
dag_run_list.append(dag_run) dag_run_list.append(dag_run)
...@@ -187,21 +187,10 @@ class BaseDAG: ...@@ -187,21 +187,10 @@ class BaseDAG:
""" """
# copy template # copy template
dag_run = self.dag_run_template.copy() dag_run = self.dag_run_template.copy()
# update dag_group_run
# update dag_group_run info dag_run = override_common_keys(dag_run, dag_group_run)
for k, v in dag_group_run.items(): # update dag_run
assert ( dag_run = override_common_keys(dag_run, dag_run_kwargs)
k in self.dag_run_template.keys()
), f"{k} not in {self.dag_run_template.keys()}"
dag_run[k] = v
# update dag_run info
for k, v in dag_run_kwargs.items():
assert (
k in self.dag_run_template.keys()
), f"{k} not in {self.dag_run_template.keys()}"
dag_run[k] = v
return dag_run return dag_run
@staticmethod @staticmethod
......
...@@ -7,6 +7,7 @@ ...@@ -7,6 +7,7 @@
"priority": 1, "priority": 1,
"pmapname": "", "pmapname": "",
"dataset": "csst-msc-c9-25sqdeg-v3", "dataset": "csst-msc-c9-25sqdeg-v3",
"instrument": "MSC",
"obs_type": "WIDE", "obs_type": "WIDE",
"obs_group": "W2", "obs_group": "W2",
"obs_id": "10100232366", "obs_id": "10100232366",
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment