Skip to content

Commit

Permalink
nit fix : variable name change.
Browse files Browse the repository at this point in the history
  • Loading branch information
valayDave committed Jul 29, 2022
1 parent 4aee4ca commit 5f4d573
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions metaflow/plugins/airflow/airflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -532,10 +532,10 @@ def _visit(node, workflow, exit_node=None):

# set max active tasks here , For more info check here :
# https://airflow.apache.org/docs/apache-airflow/stable/_api/airflow/models/dag/index.html#airflow.models.dag.DAG
other_args = (
airflow_dag_args = (
{} if self.max_workers is None else dict(max_active_tasks=self.max_workers)
)
other_args["is_paused_upon_creation"] = self.is_paused_upon_creation
airflow_dag_args["is_paused_upon_creation"] = self.is_paused_upon_creation

appending_sensors = self._collect_flow_sensors()
workflow = Workflow(
Expand All @@ -547,7 +547,7 @@ def _visit(node, workflow, exit_node=None):
tags=self.tags,
file_path=self._file_path,
graph_structure=self.graph_structure,
**other_args
**airflow_dag_args
)
workflow = _visit(self.graph["start"], workflow)

Expand Down

0 comments on commit 5f4d573

Please sign in to comment.