Fix failing no-pydantic tests for OpenAI. (#41677)

The #41554 broke exclusion of openai tests when no pydantic is installed.
apache · Aug 22, 2024 · c19986c · c19986c
1 parent c2a9833
commit c19986c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -406,10 +406,11 @@ combine-as-imports = true
 "tests/providers/google/cloud/operators/vertex_ai/test_generative_model.py" = ["E402"]
 "tests/providers/google/cloud/triggers/test_vertex_ai.py" = ["E402"]
 "tests/providers/openai/hooks/test_openai.py" = ["E402"]
+"tests/providers/openai/operators/test_openai.py" = ["E402"]
+"tests/providers/openai/triggers/test_openai.py" = ["E402"]
 "tests/providers/opensearch/conftest.py" = ["E402"]
 "tests/providers/opensearch/hooks/test_opensearch.py" = ["E402"]
 "tests/providers/opensearch/operators/test_opensearch.py" = ["E402"]
-"tests/providers/openai/operators/test_openai.py" = ["E402"]
 "tests/providers/qdrant/hooks/test_qdrant.py" = ["E402"]
 "tests/providers/qdrant/operators/test_qdrant.py" = ["E402"]
 "tests/providers/snowflake/operators/test_snowflake_sql.py" = ["E402"]

diff --git a/tests/providers/openai/operators/test_openai.py b/tests/providers/openai/operators/test_openai.py
@@ -19,10 +19,11 @@
 from unittest.mock import Mock
 
 import pytest
-from openai.types.batch import Batch
 
 openai = pytest.importorskip("openai")
 
+from openai.types.batch import Batch
+
 from airflow.exceptions import TaskDeferred
 from airflow.providers.openai.operators.openai import OpenAIEmbeddingOperator, OpenAITriggerBatchOperator
 from airflow.providers.openai.triggers.openai import OpenAIBatchTrigger

diff --git a/tests/providers/openai/triggers/test_openai.py b/tests/providers/openai/triggers/test_openai.py
@@ -22,6 +22,9 @@
 from unittest import mock
 
 import pytest
+
+openai = pytest.importorskip("openai")
+
 from openai.types import Batch
 
 from airflow.providers.openai.hooks.openai import BatchStatus