From 89029f0177d10899da9c27909bd0fd05a5447f4f Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 7 Jan 2024 16:48:30 +0100
Subject: [PATCH] add patch for encoder in None test
---
tests/unit/test_layer.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 92577f34..cdf86bbc 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -120,8 +120,9 @@ class TestRouteLayer:
assert route_layer_openai.score_threshold == 0.82
def test_initialization_no_encoder(self, openai_encoder):
- route_layer_none = RouteLayer(encoder=None)
- assert route_layer_none.score_threshold == openai_encoder.score_threshold
+ with patch("semantic_router.encoders.OpenAIEncoder") as _:
+ 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):
route_layer_cohere = RouteLayer(encoder=cohere_encoder)
--
GitLab