From ccfe4b57d59640c2ad41ff1ed20f7dcc0b1dd509 Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Mon, 16 Mar 2026 00:34:47 -0700 Subject: [PATCH] [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 --- tests/llm_translation/conftest.py | 24 +++--------------------- 1 file changed, 3 insertions(+), 21 deletions(-) diff --git a/tests/llm_translation/conftest.py b/tests/llm_translation/conftest.py index 8388e7bb10..113c91f9c2 100644 --- a/tests/llm_translation/conftest.py +++ b/tests/llm_translation/conftest.py @@ -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)