mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-07 04:24:12 +00:00
[Fix] Restore unconditional importlib.reload for llm_translation conftest
The xdist-conditional reload (manual reset in xdist mode) was missing attributes that importlib.reload resets, causing Azure connection errors. The original conftest used importlib.reload unconditionally (even under xdist) and that worked on main. Restore that behavior. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.6
parent
2372427dbc
commit
ccfe4b57d5
@@ -71,27 +71,9 @@ def setup_and_teardown(event_loop): # Add event_loop as a dependency
|
||||
original_state[attr] = getattr(litellm, attr)
|
||||
|
||||
# ---- Reset to true defaults before the test ----
|
||||
worker_id = os.environ.get("PYTEST_XDIST_WORKER", None)
|
||||
if worker_id is None:
|
||||
# Single-process mode: reload for full reset
|
||||
from litellm.litellm_core_utils.logging_worker import GLOBAL_LOGGING_WORKER
|
||||
asyncio.run(GLOBAL_LOGGING_WORKER.clear_queue())
|
||||
importlib.reload(litellm)
|
||||
else:
|
||||
# xdist mode: reset globals without reload
|
||||
for attr in (
|
||||
"callbacks",
|
||||
"success_callback",
|
||||
"failure_callback",
|
||||
"_async_success_callback",
|
||||
"_async_failure_callback",
|
||||
):
|
||||
if hasattr(litellm, attr):
|
||||
setattr(litellm, attr, [])
|
||||
|
||||
for attr, default_val in _SCALAR_DEFAULTS.items():
|
||||
if hasattr(litellm, attr):
|
||||
setattr(litellm, attr, default_val)
|
||||
from litellm.litellm_core_utils.logging_worker import GLOBAL_LOGGING_WORKER
|
||||
asyncio.run(GLOBAL_LOGGING_WORKER.clear_queue())
|
||||
importlib.reload(litellm)
|
||||
|
||||
# Set the event loop from the fixture
|
||||
asyncio.set_event_loop(event_loop)
|
||||
|
||||
Reference in New Issue
Block a user