[dagster-airlift] fix master (#25290)

## Summary & Motivation This file keeps disappearing and I don't know why ## How I Tested These Changes Existing tests now pass ## Changelog NOCHANGELOG
dagster-io · Oct 15, 2024 · 6c0c2d5 · 6c0c2d5
1 parent 0df5139
commit 6c0c2d5
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/...gster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/__init__.py b/...gster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/__init__.py
diff --git a/...t/dagster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/dags.py b/...t/dagster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/dags.py
@@ -0,0 +1,33 @@
+from datetime import datetime
+
+from airflow import DAG
+from airflow.operators.python import PythonOperator
+
+
+def print_hello() -> None:
+    print("Hello")  # noqa: T201
+
+
+default_args = {
+    "owner": "airflow",
+    "depends_on_past": False,
+    "start_date": datetime(2023, 1, 1),
+    "retries": 0,
+}
+
+
+def make_print_dag(dag_id: str) -> DAG:
+    with DAG(
+        dag_id,
+        default_args=default_args,
+        schedule_interval=None,
+        is_paused_upon_creation=False,
+    ) as dag:
+        PythonOperator(task_id="print_task", python_callable=print_hello) >> PythonOperator(
+            task_id="downstream_print_task", python_callable=print_hello
+        )  # type: ignore
+        return dag
+
+
+print_dag = make_print_dag("print_dag")
+other_print_dag = make_print_dag("other_print_dag")