From 3da48cd047dd246a75310e268af2791ec919e10c Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 15 Dec 2024 14:42:42 +0400
Subject: [PATCH] fix: single utt tests
---
tests/unit/test_router.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index 067e78a9..1741865b 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -268,7 +268,7 @@ class TestSemanticRouter:
index=index,
auto_sync="local",
)
- route_layer.add(routes=[route_single_utterance])
+ route_layer.add(routes=route_single_utterance)
assert route_layer.score_threshold == openai_encoder.score_threshold
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be updated
@@ -286,7 +286,7 @@ class TestSemanticRouter:
)
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be updated
- route_layer.add(routes=[route_single_utterance])
+ route_layer.add(routes=route_single_utterance)
assert route_layer.score_threshold == openai_encoder.score_threshold
_ = route_layer("Hello")
assert len(route_layer.index.get_utterances()) == 1
--
GitLab