|
36 | 36 | from airflow.cli import cli_parser |
37 | 37 | from airflow.cli.commands import dag_command |
38 | 38 | from airflow.dag_processing.dagbag import DagBag, sync_bag_to_db |
| 39 | +from airflow.dag_processing.processor import DagFileParsingResult, DagFileProcessorProcess |
39 | 40 | from airflow.exceptions import AirflowException |
40 | 41 | from airflow.models import DagModel, DagRun |
41 | 42 | from airflow.models.dagbag import DBDagBag |
42 | 43 | from airflow.models.serialized_dag import SerializedDagModel |
43 | 44 | from airflow.providers.standard.triggers.temporal import DateTimeTrigger, TimeDeltaTrigger |
44 | 45 | from airflow.sdk import BaseOperator, task |
45 | 46 | from airflow.sdk.definitions.dag import _run_inline_trigger |
| 47 | +from airflow.sdk.execution_time.comms import _RequestFrame, _ResponseFrame |
| 48 | +from airflow.serialization.serialized_objects import DagSerialization, LazyDeserializedDAG |
46 | 49 | from airflow.triggers.base import TriggerEvent |
47 | 50 | from airflow.utils.session import create_session |
48 | 51 | from airflow.utils.state import DagRunState |
@@ -1074,10 +1077,6 @@ def test_reserialize_should_support_multiple_bundle_name_arguments(self, configu |
1074 | 1077 |
|
1075 | 1078 | @conf_vars({("core", "load_examples"): "false"}) |
1076 | 1079 | def test_reserialize_should_make_equal_hash_with_dag_processor(self, configure_dag_bundles, session): |
1077 | | - from airflow.dag_processing.processor import DagFileParsingResult, DagFileProcessorProcess |
1078 | | - from airflow.sdk.execution_time.comms import _RequestFrame, _ResponseFrame |
1079 | | - from airflow.serialization.serialized_objects import DagSerialization, LazyDeserializedDAG |
1080 | | - |
1081 | 1080 | bundles = {"bundle_reserialize": TEST_DAGS_FOLDER / "test_dag_reserialize.py"} |
1082 | 1081 | with configure_dag_bundles(bundles): |
1083 | 1082 | dag_command.dag_reserialize( |
|
0 commit comments