diff --git a/override_settings/override.py b/override_settings/override.py index 9f59747..cff5e77 100644 --- a/override_settings/override.py +++ b/override_settings/override.py @@ -1,6 +1,9 @@ import contextlib from functools import wraps -from pydantic import BaseSettings +try: + from pydantic_settings import BaseSettings +except ImportError: + from pydantic import BaseSettings def async_override_settings(settings, **overrides): diff --git a/tests/test_override.py b/tests/test_override.py index 943c1a0..ae72446 100644 --- a/tests/test_override.py +++ b/tests/test_override.py @@ -1,5 +1,8 @@ import pytest -from pydantic import BaseSettings +try: + from pydantic_settings import BaseSettings +except ImportError: + from pydantic import BaseSettings from override_settings import async_override_settings, override_settings, check_value