From e12f8ebcc9bdf8c40a23c8217b71bbcb1cc02cc7 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Mon, 6 Jan 2025 08:59:39 +0400
Subject: [PATCH] fix: vector only test
---
tests/unit/test_router.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index 4d1378c5..799a978d 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -829,7 +829,7 @@ class TestSemanticRouter:
)
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be populated
- vector = [0.1, 0.2, 0.3]
+ vector = encoder(["hello"])
query_result = route_layer(vector=vector).name
assert query_result in ["Route 1", "Route 2"]
--
GitLab