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
No related branches found
No related tags found
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.
Finish editing this message first!
Please register or to comment