diff --git a/tests/unit/proxy/management_endpoints/scim/test_scim_v2_endpoints.py b/tests/unit/proxy/management_endpoints/scim/test_scim_v2_endpoints.py index 654b5e6dd37..828af977ba5 100644 --- a/tests/unit/proxy/management_endpoints/scim/test_scim_v2_endpoints.py +++ b/tests/unit/proxy/management_endpoints/scim/test_scim_v2_endpoints.py @@ -9,13 +9,13 @@ from types import MappingProxyType from typing import Final, assert_never from unittest.mock import ANY, AsyncMock, MagicMock, call -from litellm.proxy.management_helpers.access_group_team_sync import TEAM_ADVISORY_LOCK_SQL import pytest from fastapi import FastAPI, HTTPException from httpx import ASGITransport, AsyncClient from pytest_mock import MockerFixture import litellm.proxy.management_endpoints.scim.scim_v2 as scim_v2_module +from litellm.proxy.management_helpers.access_group_team_sync import TEAM_ADVISORY_LOCK_SQL from litellm.proxy._types import ( LiteLLM_TeamTable, LiteLLM_UserTable, @@ -1116,7 +1116,6 @@ async def test_handle_existing_user_by_email_roster_add_failure_blocks_teams_wri async def test_handle_existing_user_by_email_roster_add_already_member_repairs_user_teams( mocker, roster: list[dict[str, str]], repaired: bool ): - """An already-on-roster user missing the team gets it back, but only if the roster still has them under the lock""" existing_user = mocker.MagicMock() existing_user.user_id = "uid" existing_user.user_email = "member@example.com"