Commit 77606137 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent 0632dd9d
......@@ -111,7 +111,7 @@ class BaseDAG:
for this_task in task_list:
dag_run = self.gen_dag_run(
dag_group_run=dag_group_run,
**this_task["task"],
dag_run_kwargs=this_task["task"],
)
dag_run_list.append(dag_run)
......@@ -161,19 +161,13 @@ class BaseDAG:
priority=priority,
)
def gen_dag_run(
self,
dag_group_run: dict,
**dag_run_kwargs: Any,
) -> dict:
def gen_dag_run(self, **kwargs) -> dict:
"""Generate a complete DAG run message.
Parameters
----------
dag_group_run : dict
Output from gen_dag_group_run()
**dag_run_kwargs : Any
Additional run-specific parameters
kwargs : dict
Additional keyword arguments to override.
Returns
-------
......@@ -187,10 +181,8 @@ class BaseDAG:
"""
# copy template
dag_run = self.dag_run_template.copy()
# update dag_group_run
dag_run = override_common_keys(dag_run, dag_group_run)
# update dag_run
dag_run = override_common_keys(dag_run, dag_run_kwargs)
# update values
dag_run = override_common_keys(dag_run, kwargs)
return dag_run
@staticmethod
......
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