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

add patch for encoder in None test

parent 34135011
Branches
Tags
No related merge requests found
...@@ -120,8 +120,9 @@ class TestRouteLayer: ...@@ -120,8 +120,9 @@ class TestRouteLayer:
assert route_layer_openai.score_threshold == 0.82 assert route_layer_openai.score_threshold == 0.82
def test_initialization_no_encoder(self, openai_encoder): def test_initialization_no_encoder(self, openai_encoder):
route_layer_none = RouteLayer(encoder=None) with patch("semantic_router.encoders.OpenAIEncoder") as _:
assert route_layer_none.score_threshold == openai_encoder.score_threshold route_layer_none = RouteLayer(encoder=None)
assert route_layer_none.score_threshold == openai_encoder.score_threshold
def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder): def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder):
route_layer_cohere = RouteLayer(encoder=cohere_encoder) route_layer_cohere = RouteLayer(encoder=cohere_encoder)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment