fix for 3.9

Author: rbs333

tests/integration/test_threshold_optimizer.py

@@ -1,5 +1,10 @@
+import sys
+
 import pytest
 
+if sys.version_info.major == 3 and sys.version_info.minor < 10:
+    pytest.skip("Test requires Python 3.10 or higher", allow_module_level=True)
+
 from redisvl.extensions.llmcache import SemanticCache
 from redisvl.extensions.router import Route, SemanticRouter
 from redisvl.extensions.router.schema import RoutingConfig
@@ -114,6 +119,7 @@ def test_routes_different_distance_thresholds_optimizer_default(
 def test_routes_different_distance_thresholds_optimizer_precision(
     semantic_router, routes, redis_url, test_data_optimization
 ):
+
     redis_version = semantic_router._index.client.info()["redis_version"]
     if not compare_versions(redis_version, "7.0.0"):
         pytest.skip("Not using a late enough version of Redis")
@@ -189,6 +195,10 @@ def test_optimize_threshold_cache_default(redis_url):
         distance_threshold=null_threshold,
     )
 
+    redis_version = cache.client.info()["redis_version"]
+    if not compare_versions(redis_version, "7.0.0"):
+        pytest.skip("Not using a late enough version of Redis")
+
     paris_key = cache.store(prompt="what is the capital of france?", response="paris")
     rabat_key = cache.store(prompt="what is the capital of morocco?", response="rabat")
 
@@ -213,6 +223,10 @@ def test_optimize_threshold_cache_precision(redis_url):
         distance_threshold=null_threshold,
     )
 
+    redis_version = cache.client.info()["redis_version"]
+    if not compare_versions(redis_version, "7.0.0"):
+        pytest.skip("Not using a late enough version of Redis")
+
     paris_key = cache.store(prompt="what is the capital of france?", response="paris")
     rabat_key = cache.store(prompt="what is the capital of morocco?", response="rabat")
 
@@ -237,6 +251,10 @@ def test_optimize_threshold_cache_recall(redis_url):
         distance_threshold=null_threshold,
     )
 
+    redis_version = cache.client.info()["redis_version"]
+    if not compare_versions(redis_version, "7.0.0"):
+        pytest.skip("Not using a late enough version of Redis")
+
     paris_key = cache.store(prompt="what is the capital of france?", response="paris")
     rabat_key = cache.store(prompt="what is the capital of morocco?", response="rabat")
 

tests/unit/test_threshold_optimizer_utility.py

@@ -1,12 +1,15 @@
+import sys
+
 import pytest
+
+if sys.version_info.major == 3 and sys.version_info.minor < 10:
+    pytest.skip("Test requires Python 3.10 or higher", allow_module_level=True)
+
 from ranx import evaluate
 
 from redisvl.extensions.threshold_optimizer.cache import _generate_run_cache
 from redisvl.extensions.threshold_optimizer.schema import TestData
-from redisvl.extensions.threshold_optimizer.utils import (
-    NULL_RESPONSE_KEY,
-    _format_qrels,
-)
+from redisvl.extensions.threshold_optimizer.utils import _format_qrels
 
 # Note: these tests are not intended to test ranx but to test that our data formatting for the package is correct