Skip to content
Snippets Groups Projects
Unverified Commit ce0489b5 authored by James Briggs's avatar James Briggs
Browse files

fix: threshold values in test

parent 1772926d
No related branches found
No related tags found
No related merge requests found
...@@ -542,7 +542,7 @@ class TestRouteLayer: ...@@ -542,7 +542,7 @@ class TestRouteLayer:
route_layer = RouteLayer( route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, index=index_cls() 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( def test_with_multiple_routes_passing_threshold(
self, openai_encoder, routes, index_cls self, openai_encoder, routes, index_cls
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment