diff --git a/setup.cfg b/setup.cfg index 2cb536f5..00798242 100644 --- a/setup.cfg +++ b/setup.cfg @@ -41,6 +41,7 @@ dev = pytest==7.4.2 python-dotenv==1.0.0 ruff==0.0.292 + pytest-asyncio==0.23.5 [mypy] ignore_missing_imports = true diff --git a/tests/test_async_chat_complete.py b/tests/test_async_chat_complete.py index 10325e2d..b65625cd 100644 --- a/tests/test_async_chat_complete.py +++ b/tests/test_async_chat_complete.py @@ -31,7 +31,7 @@ def get_configs(folder_path) -> List[Dict[str, Any]]: class TestChatCompletions: client = AsyncPortkey parametrize = pytest.mark.parametrize("client", [client], ids=["strict"]) - models = read_json_file("./models.json") + models = read_json_file("./tests/models.json") def get_metadata(self): return { diff --git a/tests/test_async_complete.py b/tests/test_async_complete.py index cb20cb20..21355fef 100644 --- a/tests/test_async_complete.py +++ b/tests/test_async_complete.py @@ -31,7 +31,7 @@ def get_configs(folder_path) -> List[Dict[str, Any]]: class TestChatCompletions: client = AsyncPortkey parametrize = pytest.mark.parametrize("client", [client], ids=["strict"]) - models = read_json_file("./models.json") + models = read_json_file("./tests/models.json") def get_metadata(self): return {