From e6c328ab2d5ae2cfa70afe05bbb18ba31b5adcfc Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sun, 10 Nov 2024 01:26:27 +0100
Subject: [PATCH] chore: lint
---
semantic_router/index/pinecone.py | 2 --
tests/unit/test_sync.py | 11 +++--------
2 files changed, 3 insertions(+), 10 deletions(-)
diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py
index cfe5ce53..2c4c4961 100644
--- a/semantic_router/index/pinecone.py
+++ b/semantic_router/index/pinecone.py
@@ -488,8 +488,6 @@ class PineconeIndex(BaseIndex):
self.dimensions = self.dimensions or len(embeddings[0])
self.index = self._init_index(force_create=True)
- print([(r, u, fs, m) for r, u, fs, m in zip(routes, utterances, function_schemas, metadata_list)])
-
vectors_to_upsert = [
PineconeRecord(
values=vector,
diff --git a/tests/unit/test_sync.py b/tests/unit/test_sync.py
index c18f533d..1e4b72c1 100644
--- a/tests/unit/test_sync.py
+++ b/tests/unit/test_sync.py
@@ -5,7 +5,6 @@ import pytest
import time
from typing import Optional
from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder
-from semantic_router.index.local import LocalIndex
from semantic_router.index.pinecone import PineconeIndex
from semantic_router.index.qdrant import QdrantIndex
from semantic_router.layer import RouteLayer
@@ -38,7 +37,7 @@ def init_index(
index_cls,
dimensions: Optional[int] = None,
namespace: Optional[str] = "",
- sync: str = None,
+ sync: Optional[str] = None,
):
"""We use this function to initialize indexes with different names to avoid
issues during testing.
@@ -192,9 +191,7 @@ class TestRouteLayer:
)
def test_initialization(self, openai_encoder, routes, index_cls):
index = init_index(index_cls)
- _ = RouteLayer(
- encoder=openai_encoder, routes=routes, top_k=10, index=index
- )
+ _ = RouteLayer(encoder=openai_encoder, routes=routes, top_k=10, index=index)
@pytest.mark.skipif(
os.environ.get("PINECONE_API_KEY") is None, reason="Pinecone API key required"
@@ -227,9 +224,7 @@ class TestRouteLayer:
)
def test_utterance_diff(self, openai_encoder, routes, routes_2, index_cls):
index = init_index(index_cls)
- _ = RouteLayer(
- encoder=openai_encoder, routes=routes, top_k=10, index=index
- )
+ _ = RouteLayer(encoder=openai_encoder, routes=routes, top_k=10, index=index)
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be populated
route_layer_2 = RouteLayer(
--
GitLab