fix: threshold values in test

aurelio-labs · Jul 19, 2024 · ce0489b · ce0489b
1 parent 1772926
commit ce0489b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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