From 9f60017f02c0dbdf3c835e0575fe220ec2bb1aff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Pedersen?= <andrped94@gmail.com>
Date: Wed, 13 Mar 2024 18:57:56 +0100
Subject: [PATCH] Add alpha selection test for Hybrid route layer
---
tests/unit/test_hybrid_layer.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/unit/test_hybrid_layer.py b/tests/unit/test_hybrid_layer.py
index 52eb6b3e..29643e95 100644
--- a/tests/unit/test_hybrid_layer.py
+++ b/tests/unit/test_hybrid_layer.py
@@ -81,12 +81,13 @@ sparse_encoder.fit(["The quick brown fox", "jumps over the lazy dog", "Hello, wo
class TestHybridRouteLayer:
def test_initialization(self, openai_encoder, routes):
route_layer = HybridRouteLayer(
- encoder=openai_encoder, sparse_encoder=sparse_encoder, routes=routes, top_k=10,
+ encoder=openai_encoder, sparse_encoder=sparse_encoder, routes=routes, top_k=10, alpha=0.8,
)
assert route_layer.index is not None and route_layer.categories is not None
assert openai_encoder.score_threshold == 0.82
assert route_layer.score_threshold == 0.82
assert route_layer.top_k == 10
+ assert route_layer.alpha == 0.8
assert len(route_layer.index) == 5
assert len(set(route_layer.categories)) == 2
--
GitLab