diff --git a/wfcommons/wfbench/translator/airflow.py b/wfcommons/wfbench/translator/airflow.py index e8c9d9db..47570d8d 100644 --- a/wfcommons/wfbench/translator/airflow.py +++ b/wfcommons/wfbench/translator/airflow.py @@ -41,8 +41,6 @@ def __init__(self, self.script = f""" from __future__ import annotations -import os -from datetime import datetime from airflow.models.dag import DAG from airflow.operators.bash import BashOperator @@ -64,9 +62,6 @@ def translate(self, output_folder: pathlib.Path, name: Optional[str] = None) -> with DAG( "{name}", description="airflow translation of a wfcommons instance", - schedule="0 0 * * *", - start_date=datetime(2021, 1, 1), - catchup=False, tags=["wfcommons"], ) as dag: """ @@ -77,12 +72,10 @@ def translate(self, output_folder: pathlib.Path, name: Optional[str] = None) -> self.script += f""" {self._sanitize_varname(task.task_id)} = BashOperator( task_id="{task.task_id}", - depends_on_past=False, bash_command='{self.task_commands[task.task_id]}', - env={{"AIRFLOW_HOME": os.environ["AIRFLOW_HOME"]}}, - retries=3, ) """ + for task in self.tasks.values(): # Comma-separated list of the task's parents parents = ", ".join(map(self._sanitize_varname, self.task_parents[task.task_id]))