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

This commit is contained in:
Krrish Dholakia 2024-01-13 14:23:04 +05:30
parent d41b84ec40
commit 9e6451e8f9

View file

@ -29,6 +29,10 @@ def setup_and_teardown():
# from litellm import Router, completion, aembedding, acompletion, embedding
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):
# Separate tests in 'test_amazing_proxy_custom_logger.py' and other tests