test(alerts): support budget regression on Python 3.12

This commit is contained in:
moe-berri 2026-11-06 16:09:36 +08:00
parent da2a44bf1f
commit 2cc807cec4

View file

@ -68,7 +68,7 @@ class _BudgetDatabase:
self.lease_rollback_cancelled = False
@asynccontextmanager
async def tx(self, *, max_wait: timedelta, timeout: timedelta) -> AsyncGenerator[AnalyticsTransaction]:
async def tx(self, *, max_wait: timedelta, timeout: timedelta) -> AsyncGenerator[AnalyticsTransaction, None]:
transaction: Final = _BudgetTransaction(self, self.transactions)
self.transactions += 1
self.active += 1