From ce0489b50872051d80c32d2ace39b528ff2c1d4e Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
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 @@ 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
--
GitLab