Skip to content

Commit fbe3298

Browse files
wjddn279eladkal
authored andcommitted
fix import location
1 parent fa8b5c8 commit fbe3298

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

airflow-core/tests/unit/cli/commands/test_dag_command.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,13 +36,16 @@
3636
from airflow.cli import cli_parser
3737
from airflow.cli.commands import dag_command
3838
from airflow.dag_processing.dagbag import DagBag, sync_bag_to_db
39+
from airflow.dag_processing.processor import DagFileParsingResult, DagFileProcessorProcess
3940
from airflow.exceptions import AirflowException
4041
from airflow.models import DagModel, DagRun
4142
from airflow.models.dagbag import DBDagBag
4243
from airflow.models.serialized_dag import SerializedDagModel
4344
from airflow.providers.standard.triggers.temporal import DateTimeTrigger, TimeDeltaTrigger
4445
from airflow.sdk import BaseOperator, task
4546
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
4649
from airflow.triggers.base import TriggerEvent
4750
from airflow.utils.session import create_session
4851
from airflow.utils.state import DagRunState
@@ -1074,10 +1077,6 @@ def test_reserialize_should_support_multiple_bundle_name_arguments(self, configu
10741077

10751078
@conf_vars({("core", "load_examples"): "false"})
10761079
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-
10811080
bundles = {"bundle_reserialize": TEST_DAGS_FOLDER / "test_dag_reserialize.py"}
10821081
with configure_dag_bundles(bundles):
10831082
dag_command.dag_reserialize(

0 commit comments

Comments
 (0)