fix(router.py): fix pydantic object logic

This commit is contained in:
Krrish Dholakia 2024-04-03 21:57:19 -07:00
parent acdb3a7f7e
commit a4a8129a13
2 changed files with 16 additions and 8 deletions

View file

@ -181,7 +181,7 @@ def test_weighted_selection_router_tpm_as_router_param():
pytest.fail(f"Error occurred: {e}")
test_weighted_selection_router_tpm_as_router_param()
# test_weighted_selection_router_tpm_as_router_param()
def test_weighted_selection_router_rpm_as_router_param():
@ -433,7 +433,7 @@ def test_usage_based_routing():
selection_counts[response["model"]] += 1
# print("selection counts", selection_counts)
print("selection counts", selection_counts)
total_requests = sum(selection_counts.values())