diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index cdf86bbcd5c6b0d19dd150ca913a249dcbe2e37d..6e2ac317000a20f10d45cd671227e4831057e2da 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -120,9 +120,9 @@ class TestRouteLayer: assert route_layer_openai.score_threshold == 0.82 def test_initialization_no_encoder(self, openai_encoder): - with patch("semantic_router.encoders.OpenAIEncoder") as _: - route_layer_none = RouteLayer(encoder=None) - assert route_layer_none.score_threshold == openai_encoder.score_threshold + os.environ["OPENAI_API_KEY"] = "test_api_key" + 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)