From ce0489b50872051d80c32d2ace39b528ff2c1d4e Mon Sep 17 00:00:00 2001 From: James Briggs Date: Fri, 19 Jul 2024 15:45:33 +0800 Subject: [PATCH] fix: threshold values in test --- tests/unit/test_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 74c069fa..05979cd2 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -542,7 +542,7 @@ def test_get_thresholds(self, openai_encoder, routes, index_cls): route_layer = RouteLayer( encoder=openai_encoder, routes=routes, index=index_cls() ) - assert route_layer.get_thresholds() == {"Route 1": 0.82, "Route 2": 0.82} + assert route_layer.get_thresholds() == {"Route 1": 0.3, "Route 2": 0.3} def test_with_multiple_routes_passing_threshold( self, openai_encoder, routes, index_cls