fix: test max agg method

aurelio-labs · Jan 10, 2025 · 19169e7 · 19169e7
1 parent 351f432
commit 19169e7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
@@ -772,6 +772,7 @@ def test_query_and_classification(self, routes, index_cls, encoder_cls, router_c
             routes=routes,
             index=index,
             auto_sync="local",
+            aggregation="max",
         )
 
         @retry(max_retries=RETRY_COUNT, delay=PINECONE_SLEEP)
@@ -789,6 +790,7 @@ def test_query_filter(self, routes, index_cls, encoder_cls, router_cls):
             routes=routes,
             index=index,
             auto_sync="local",
+            aggregation="max",
         )
 
         @retry(max_retries=RETRY_COUNT, delay=PINECONE_SLEEP)
@@ -851,6 +853,7 @@ def test_query_with_vector(self, routes, index_cls, encoder_cls, router_cls):
             routes=routes,
             index=index,
             auto_sync="local",
+            aggregation="max",
         )
         # create vectors
         vector = encoder(["hello"])