[AIRFLOW-XXX] Fix log handler test

(cherry picked from commit 606b006) Signed-off-by: Bolke de Bruin <[email protected]>
soundcloud · Dec 8, 2017 · 5b25b06 · 5b25b06
1 parent 2205f49
commit 5b25b06
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/utils/test_log_handlers.py b/tests/utils/test_log_handlers.py
@@ -22,6 +22,7 @@
 from airflow.models import TaskInstance, DAG, DagRun
 from airflow.config_templates.airflow_local_settings import DEFAULT_LOGGING_CONFIG
 from airflow.operators.dummy_operator import DummyOperator
+from airflow.operators.python_operator import PythonOperator
 from airflow.settings import Session
 from airflow.utils.log.logging_mixin import set_context
 from airflow.utils.log.file_task_handler import FileTaskHandler