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

tweaks

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