fix: single utt tests

aurelio-labs · Dec 15, 2024 · 3da48cd · 3da48cd
1 parent 8f0f65b
commit 3da48cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
@@ -268,7 +268,7 @@ def test_add_single_utterance(
             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 @@ def test_init_and_add_single_utterance(
         )
         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