diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py index 3d476b1b248732460b1fcb98bb4a77f2907bb67a..aa0c83fa4ca2021d59ca60e9f507380d6d86652c 100644 --- a/tests/unit/test_router.py +++ b/tests/unit/test_router.py @@ -16,7 +16,7 @@ from semantic_router.route import Route from platform import python_version -PINECONE_SLEEP = 12 +PINECONE_SLEEP = 10 def mock_encoder_call(utterances): @@ -137,7 +137,9 @@ def openai_encoder(mocker): mocker.patch.object(OpenAIEncoder, "acall", side_effect=async_mock_encoder_call) # Create and return the mocked encoder - encoder = OpenAIEncoder(name="text-embedding-3-small") + encoder = OpenAIEncoder( + name="text-embedding-3-small" + ) return encoder @@ -267,7 +269,7 @@ class TestIndexEncoders: top_k=10, ) if index_cls is PineconeIndex: - time.sleep(PINECONE_SLEEP) # allow for index to be populated + time.sleep(PINECONE_SLEEP*2) # allow for index to be populated if isinstance(route_layer, HybridRouter): assert (