From 608af1b1258da2d4d3e4c4d12f0d7ef2fc9ccbec Mon Sep 17 00:00:00 2001
From: zahid-syed <zahid.s2618@gmail.com>
Date: Wed, 20 Mar 2024 10:38:19 -0400
Subject: [PATCH] linting
---
tests/unit/test_layer.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index a54891f3..fa5482ab 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder
from semantic_router.layer import LayerConfig, RouteLayer
from semantic_router.llms.base import BaseLLM
from semantic_router.route import Route
-from semantic_router.index.pinecone import PineconeIndex
def mock_encoder_call(utterances):
@@ -215,12 +214,12 @@ class TestRouteLayer:
route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
query_result = route_layer(text="Hello").name
assert query_result in ["Route 1", "Route 2"]
-
+
def test_query_filter(self, openai_encoder, routes):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
assert query_result in ["Route 1"]
-
+
# def test_query_filter_openai_index(self, openai_encoder, routes):
# route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=PineconeIndex)
# query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
--
GitLab