diff --git a/tests/integration/database/test_proxy_database_lifespan.py b/tests/integration/database/test_proxy_database_lifespan.py index c79c0ba4b35..e24f565c437 100644 --- a/tests/integration/database/test_proxy_database_lifespan.py +++ b/tests/integration/database/test_proxy_database_lifespan.py @@ -10,7 +10,6 @@ from typing import Final, Literal, TypeAlias import psutil import pytest -from litellm.rust_bridge._native import NativeDatabase from tests.integration._support.client import eventually from tests.integration._support.database import read_rows, scratch_database @@ -80,7 +79,7 @@ async def _exercise_lifespans(mode: Mode) -> None: async with proxy_server.proxy_startup_event(proxy_server.app) as state: database: Final = state["database"] assert database is not None - assert isinstance(database.native, NativeDatabase) + assert database.native is None assert await database.prisma.writer_db.query_raw("SELECT 42 AS answer") == [{"answer": 42}] assert await database.prisma.read_db.query_raw("SELECT 43 AS answer") == [{"answer": 43}] assert psutil.Process().children(recursive=True)