phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(conftest.py): create an event loop if one isn't made

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-13 14:23:04 +05:30
parent d41b84ec40
commit 9e6451e8f9
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/conftest.py
View file

@ -29,6 +29,10 @@ def setup_and_teardown():
# from litellm import Router, completion, aembedding, acompletion, embedding # from litellm import Router, completion, aembedding, acompletion, embedding
yield yield
# Teardown code (executes after the yield point)
loop.close() # Close the loop created earlier
asyncio.set_event_loop(None) # Remove the reference to the loop
def pytest_collection_modifyitems(config, items): def pytest_collection_modifyitems(config, items):
# Separate tests in 'test_amazing_proxy_custom_logger.py' and other tests # Separate tests in 'test_amazing_proxy_custom_logger.py' and other tests