diff --git a/integration_tests/test_pydantic.py b/integration_tests/test_pydantic.py index 2ceff6a39..2661baffe 100644 --- a/integration_tests/test_pydantic.py +++ b/integration_tests/test_pydantic.py @@ -1,3 +1,5 @@ +import importlib.util + import pytest import requests @@ -5,12 +7,7 @@ BASE_URL = "http://127.0.0.1:8080" -try: - import pydantic - - _HAS_PYDANTIC = True -except ImportError: - _HAS_PYDANTIC = False +_HAS_PYDANTIC = importlib.util.find_spec("pydantic") is not None pytestmark = pytest.mark.skipif(not _HAS_PYDANTIC, reason="pydantic not installed") diff --git a/robyn/openapi.py b/robyn/openapi.py index a8f5981d9..7383a3daf 100644 --- a/robyn/openapi.py +++ b/robyn/openapi.py @@ -9,9 +9,9 @@ from pathlib import Path from typing import Any, Callable, Dict, List, Optional, Tuple, TypedDict, is_typeddict +from robyn.pydantic_support import get_pydantic_openapi_schema, is_pydantic_model from robyn.responses import html from robyn.robyn import QueryParams, Response -from robyn.pydantic_support import get_pydantic_openapi_schema, is_pydantic_model from robyn.types import Body, JsonBody _logger = logging.getLogger(__name__)