test: fix linting testing

This commit is contained in:
Krrish Dholakia 2023-10-19 18:40:33 -07:00
parent 6e7dd8d64f
commit 494b9bbc62
3 changed files with 6 additions and 6 deletions

View file

@ -221,5 +221,5 @@ class Router:
# ------------
# Update usage
# ------------
self.cache.increment(tpm_key, total_tokens)
self.cache.increment(rpm_key, 1)
self.increment(tpm_key, total_tokens)
self.increment(rpm_key, 1)