diff --git a/experiment-tracker/tests/integration/conftest.py b/experiment-tracker/tests/integration/conftest.py index bece8107b..52393e605 100644 --- a/experiment-tracker/tests/integration/conftest.py +++ b/experiment-tracker/tests/integration/conftest.py @@ -2,7 +2,7 @@ import png import pytest -from pydantic import BaseSettings +from pydantic_settings import BaseSettings class TestSettings(BaseSettings): diff --git a/experiment-tracker/tests/requirements-test.txt b/experiment-tracker/tests/requirements-test.txt index b7e11343b..415e50750 100644 --- a/experiment-tracker/tests/requirements-test.txt +++ b/experiment-tracker/tests/requirements-test.txt @@ -1,5 +1,6 @@ pytest pydantic +pydantic-settings pypng aiohttp pytest-asyncio