diff --git a/tests/unit/test_hybrid_layer.py b/tests/unit/test_hybrid_layer.py index 52eb6b3e65d2a25b5a33f51454f5397aedf630a9..29643e958dd029f93e8b99c8ae99635598811ce9 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