From 86676142c99b933dfbf757e36536bfa6c8a4f00d Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Fri, 23 Jan 2026 22:33:00 -0800 Subject: [PATCH] Fixing failing tests --- tests/local_testing/test_add_update_models.py | 2 ++ tests/test_proxy_server_non_root.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_add_update_models.py b/tests/local_testing/test_add_update_models.py index 6d7ae597b5..4b5ec95d3f 100644 --- a/tests/local_testing/test_add_update_models.py +++ b/tests/local_testing/test_add_update_models.py @@ -16,6 +16,8 @@ sys.path.insert( ) # Adds the parent directory to the system path import pytest, logging, asyncio import litellm +import litellm.proxy +import litellm.proxy.proxy_server from litellm.proxy.management_endpoints.model_management_endpoints import ( add_new_model, update_model, diff --git a/tests/test_proxy_server_non_root.py b/tests/test_proxy_server_non_root.py index 6a73b509df..03ef80c705 100644 --- a/tests/test_proxy_server_non_root.py +++ b/tests/test_proxy_server_non_root.py @@ -1,6 +1,7 @@ from unittest.mock import patch import pytest -@pytest.mark.skip(reason="Very Flaky in CI, will debug later") +import litellm.proxy +import litellm.proxy.proxy_server def test_restructure_ui_html_files_skipped_in_non_root(monkeypatch): """ Test that _restructure_ui_html_files is SKIPPED when: @@ -36,7 +37,6 @@ def test_restructure_ui_html_files_skipped_in_non_root(monkeypatch): # Verify it was NOT called mock_restructure.assert_not_called() -@pytest.mark.skip(reason="Very Flaky in CI, will debug later") def test_restructure_ui_html_files_NOT_skipped_locally(monkeypatch): """ Test that _restructure_ui_html_files is NOT skipped for local development