Fix imports in example DAG

apache · Sep 13, 2021 · 8897d1a · 8897d1a
1 parent 02be21a
commit 8897d1a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/airflow/example_dags/example_workday_timetable.py b/airflow/example_dags/example_workday_timetable.py
@@ -18,16 +18,13 @@
 
 """Example DAG demostrating how to implement a custom timetable for a DAG."""
 
-from plugins.workday import AfterWorkdayTimetable
-
 # [START howto_timetable]
 from airflow import DAG
+from airflow.example_dags.plugins.workday import AfterWorkdayTimetable
 from airflow.operators.dummy import DummyOperator
 
-# [START howto_timetable_example_dag]
 with DAG(timetable=AfterWorkdayTimetable(), tags=["example", "timetable"]) as dag:
     DummyOperator(task_id="run_this")
-# [END howto_timetable_example_dag]
 
 if __name__ == "__main__":
     dag.cli()

diff --git a/docs/apache-airflow/howto/timetable.rst b/docs/apache-airflow/howto/timetable.rst
@@ -88,11 +88,13 @@ Next, we'll start putting code into ``AfterWorkdayTimetable``. After the
 implementation is finished, we should be able to use the timetable in our DAG
 file:
 
-.. exampleinclude:: /../../airflow/example_dags/example_workday_timetable.py
-    :language: python
-    :dedent: 4
-    :start-after: [START howto_timetable_example_dag]
-    :end-before: [END howto_timetable_example_dag]
+.. code-block:: python
+
+    from airflow import DAG
+
+
+    with DAG(timetable=AfterWorkdayTimetable(), tags=["example", "timetable"]) as dag:
+        ...
 
 
 Define Scheduling Logic