diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 74c069fa0ba86481d0142b0b8477c5b3758f837e..05979cd22d05c5ab696c984b2632c572e5e2971e 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -542,7 +542,7 @@ class TestRouteLayer: 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