diff --git a/providers/apache/beam/tests/system/apache/beam/example_beam.py b/providers/apache/beam/tests/system/apache/beam/example_beam.py index f58937eaf9abe..77464f157e564 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_beam.py +++ b/providers/apache/beam/tests/system/apache/beam/example_beam.py @@ -21,14 +21,15 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( GCS_INPUT, GCS_JAR_DIRECT_RUNNER_BUCKET_NAME, GCS_JAR_DIRECT_RUNNER_OBJECT_NAME, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator with models.DAG( diff --git a/providers/apache/beam/tests/system/apache/beam/example_beam_java_flink.py b/providers/apache/beam/tests/system/apache/beam/example_beam_java_flink.py index d001db8abb084..f34b8bea2166f 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_beam_java_flink.py +++ b/providers/apache/beam/tests/system/apache/beam/example_beam_java_flink.py @@ -21,14 +21,15 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( GCS_INPUT, GCS_JAR_FLINK_RUNNER_BUCKET_NAME, GCS_JAR_FLINK_RUNNER_OBJECT_NAME, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator with models.DAG( diff --git a/providers/apache/beam/tests/system/apache/beam/example_beam_java_spark.py b/providers/apache/beam/tests/system/apache/beam/example_beam_java_spark.py index 8d10c437376af..a55225670fbfe 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_beam_java_spark.py +++ b/providers/apache/beam/tests/system/apache/beam/example_beam_java_spark.py @@ -21,14 +21,15 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( GCS_INPUT, GCS_JAR_SPARK_RUNNER_BUCKET_NAME, GCS_JAR_SPARK_RUNNER_OBJECT_NAME, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator with models.DAG( diff --git a/providers/apache/beam/tests/system/apache/beam/example_go.py b/providers/apache/beam/tests/system/apache/beam/example_go.py index d5b94a193a08b..5e2b63ea705bc 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_go.py +++ b/providers/apache/beam/tests/system/apache/beam/example_go.py @@ -21,9 +21,7 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( DEFAULT_ARGS, GCP_PROJECT_ID, GCS_GO, @@ -32,6 +30,9 @@ GCS_TMP, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration with models.DAG( diff --git a/providers/apache/beam/tests/system/apache/beam/example_go_dataflow.py b/providers/apache/beam/tests/system/apache/beam/example_go_dataflow.py index 8327abc9ee274..550ae1ea4031c 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_go_dataflow.py +++ b/providers/apache/beam/tests/system/apache/beam/example_go_dataflow.py @@ -21,9 +21,7 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( DEFAULT_ARGS, GCP_PROJECT_ID, GCS_GO_DATAFLOW_ASYNC, @@ -32,6 +30,9 @@ GCS_TMP, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator from airflow.providers.google.cloud.hooks.dataflow import DataflowJobStatus from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration from airflow.providers.google.cloud.sensors.dataflow import DataflowJobStatusSensor diff --git a/providers/apache/beam/tests/system/apache/beam/example_java_dataflow.py b/providers/apache/beam/tests/system/apache/beam/example_java_dataflow.py index 1bc3803ba0b71..66a845fc266cd 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_java_dataflow.py +++ b/providers/apache/beam/tests/system/apache/beam/example_java_dataflow.py @@ -21,9 +21,7 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( GCS_JAR_DATAFLOW_RUNNER_BUCKET_NAME, GCS_JAR_DATAFLOW_RUNNER_OBJECT_NAME, GCS_OUTPUT, @@ -31,6 +29,9 @@ GCS_TMP, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator with models.DAG( diff --git a/providers/apache/beam/tests/system/apache/beam/example_python.py b/providers/apache/beam/tests/system/apache/beam/example_python.py index 1b67fbe2bdd40..cac2333afc2fe 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_python.py +++ b/providers/apache/beam/tests/system/apache/beam/example_python.py @@ -21,9 +21,7 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( DEFAULT_ARGS, GCP_PROJECT_ID, GCS_OUTPUT, @@ -32,6 +30,9 @@ GCS_TMP, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration with models.DAG( diff --git a/providers/apache/beam/tests/system/apache/beam/example_python_async.py b/providers/apache/beam/tests/system/apache/beam/example_python_async.py index 5dcc43afd4497..5ec1bed622ea4 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_python_async.py +++ b/providers/apache/beam/tests/system/apache/beam/example_python_async.py @@ -21,9 +21,7 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( DEFAULT_ARGS, GCP_PROJECT_ID, GCS_OUTPUT, @@ -32,6 +30,9 @@ GCS_TMP, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration with models.DAG( diff --git a/providers/apache/beam/tests/system/apache/beam/example_python_dataflow.py b/providers/apache/beam/tests/system/apache/beam/example_python_dataflow.py index 3a15f98417a1c..eea6410839b11 100644 --- a/providers/apache/beam/tests/system/apache/beam/example_python_dataflow.py +++ b/providers/apache/beam/tests/system/apache/beam/example_python_dataflow.py @@ -21,9 +21,7 @@ from __future__ import annotations -from airflow import models -from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator -from airflow.providers.apache.beam.tests.system.apache.beam.utils import ( +from providers.apache.beam.tests.system.apache.beam.utils import ( DEFAULT_ARGS, GCP_PROJECT_ID, GCS_OUTPUT, @@ -32,6 +30,9 @@ GCS_TMP, START_DATE, ) + +from airflow import models +from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator from airflow.providers.google.cloud.hooks.dataflow import DataflowJobStatus from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration from airflow.providers.google.cloud.sensors.dataflow import DataflowJobStatusSensor diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage.py b/providers/openlineage/tests/system/openlineage/example_openlineage.py index 11dbf6d89b47f..8e632d2c2dcbc 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage.py @@ -19,8 +19,9 @@ import os from datetime import datetime +from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator + from airflow import DAG -from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_sensor.py b/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_sensor.py index ecb5d20a643ff..44fecc85a35e5 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_sensor.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_sensor.py @@ -19,9 +19,10 @@ import os from datetime import datetime, timedelta +from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator + from airflow import DAG from airflow.models import Variable -from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator from airflow.providers.standard.sensors.time_delta import TimeDeltaSensor diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py b/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py index 89aa7aee56cd6..4ebc9fb7893f1 100644 --- a/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py +++ b/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py @@ -29,8 +29,8 @@ from airflow.providers.google.cloud.operators.gcs import GCSCreateBucketOperator, GCSDeleteBucketOperator from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator -from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.utils.trigger_rule import TriggerRule +from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from providers.tests.system.google import DEFAULT_GCP_SYSTEM_TEST_PROJECT_ID