shahar1 commented on code in PR #38673:
URL: https://github.com/apache/airflow/pull/38673#discussion_r1558240411
##########
tests/providers/google/cloud/operators/test_automl.py:
##########
@@ -21,11 +21,16 @@
from unittest import mock
import pytest
-from google.api_core.gapic_v1.method import DEFAULT
-from google.cloud.automl_v1beta1 import BatchPredictResult, Dataset, Model,
PredictResponse
-from airflow.providers.google.cloud.hooks.automl import CloudAutoMLHook
-from airflow.providers.google.cloud.operators.automl import (
+# For no Pydantic environment, we need to skip the tests
+pytest.importorskip("google.cloud.aiplatform_v1")
+
+from google.api_core.gapic_v1.method import DEFAULT # noqa: E402
+from google.cloud.automl_v1beta1 import BatchPredictResult, Dataset, Model,
PredictResponse # noqa: E402
+
+from airflow.exceptions import AirflowProviderDeprecationWarning # noqa: E402
+from airflow.providers.google.cloud.hooks.automl import CloudAutoMLHook #
noqa: E402
+from airflow.providers.google.cloud.operators.automl import ( # noqa: E402
Review Comment:
Instead of having these `#noqa: E402`, you could try adding a line in
`pyproject.toml` (around L356)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
