diff --git a/pipeline/dags/school_data_pipeline.py b/pipeline/dags/school_data_pipeline.py index 6829819..12551c1 100644 --- a/pipeline/dags/school_data_pipeline.py +++ b/pipeline/dags/school_data_pipeline.py @@ -15,12 +15,16 @@ from __future__ import annotations from datetime import datetime, timedelta from airflow import DAG -from airflow.providers.standard.operators.bash import BashOperator from airflow.utils.task_group import TaskGroup +try: + from airflow.providers.standard.operators.bash import BashOperator +except ImportError: + from airflow.operators.bash import BashOperator + PIPELINE_DIR = "/opt/pipeline" -MELTANO_BIN = f"{PIPELINE_DIR}/.venv/bin/meltano" -DBT_BIN = f"{PIPELINE_DIR}/.venv/bin/dbt" +MELTANO_BIN = "meltano" +DBT_BIN = "dbt" default_args = { "owner": "school-compare",