Update tests/test_litellm/integrations/test_prometheus_double_counting.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Harshit Jain 2026-03-07 04:03:24 +05:30 • committed by GitHub
parent 59f5365127
commit 42861114a7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -167,7 +167,7 @@ def test_callback_deduplication():
# Since we added "prometheus" string twice, it might initialize it twice, but replacing
# in-place should lead to correct count matching replaced. Actually the old buggy code
# wouldn't remove "prometheus". The new one replaces it.
assert len(prometheus_instances) > 0
assert len(prometheus_instances) == 1
finally:
# Restore
litellm.callbacks = old_callbacks