test: fix linting testing

This commit is contained in:
Krrish Dholakia 2023-10-19 18:40:33 -07:00
parent 4198901a2d
commit 04f8840a92
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)