kill type safe warnings

Author: rbs333

redisvl/utils/optimize/router.py

@@ -18,9 +18,9 @@ def _generate_run_router(test_data: List[LabeledData], router: SemanticRouter) -
         run_dict[td.id] = {}
         route_match = router(td.query)
         if route_match and route_match.name == td.query_match:
-            run_dict[td.id][td.query_match] = 1
+            run_dict[td.id][td.query_match] = np.int64(1)
         else:
-            run_dict[td.id][NULL_RESPONSE_KEY] = 1
+            run_dict[td.id][NULL_RESPONSE_KEY] = np.int64(1)
 
     return Run(run_dict)
 

redisvl/utils/optimize/utils.py

@@ -1,5 +1,6 @@
 from typing import List
 
+import numpy as np
 from ranx import Qrels
 
 from redisvl.utils.optimize.schema import LabeledData
@@ -13,10 +14,10 @@ def _format_qrels(test_data: List[LabeledData]) -> Qrels:
 
     for td in test_data:
         if td.query_match:
-            qrels_dict[td.id] = {td.query_match: 1}
+            qrels_dict[td.id] = {td.query_match: np.int64(1)}
         else:
             # This is for capturing true negatives from test set
-            qrels_dict[td.id] = {NULL_RESPONSE_KEY: 1}
+            qrels_dict[td.id] = {NULL_RESPONSE_KEY: np.int64(1)}
 
     return Qrels(qrels_dict)
 

schemas/semantic_router.yaml

@@ -1,4 +1,4 @@
-name: test-router-01JSHERM5V4G94GN3W68XB45PK
+name: test-router-01JSHK4MJ79HH51PS6WEK6M9MF
 routes:
 - name: greeting
   references:

tests/integration/test_semantic_router.py

@@ -475,7 +475,10 @@ def test_add_delete_route_references(semantic_router):
     assert len(router_dict["routes"][1]["references"]) == 0
 
 
-def test_from_existing(redis_url, routes):
+def test_from_existing(client, redis_url, routes):
+    if not compare_versions(client.info()["redis_version"], "7.0.0"):
+        pytest.skip("Not using a late enough version of Redis")
+
     # connect separately
     router = SemanticRouter(
         name=f"test-router-{str(ULID())}",