Merge pull request #632 from aurelio-labs/josh/pytest-deps

feat: updated toml to remove torch in main

Author: jamescalam

pyproject.toml

@@ -19,8 +19,7 @@ dependencies = [
     "tornado>=6.4.2,<7",
     "urllib3>=1.26,<3",
     "litellm>=1.61.3",
-    "openai>=1.10.0,<2.0.0",
-    "torch>=2.6.0 ; python_version < '3.13'",
+    "openai>=1.10.0,<2.0.0"
 ]
 
 [project.optional-dependencies]
@@ -29,14 +28,16 @@ local = [
     "transformers>=4.36.2 ; python_version < '3.13'",
     "tokenizers>=0.19 ; python_version < '3.13'",
     "llama-cpp-python>=0.2.28,<0.2.86 ; python_version < '3.13'",
-    "sentence-transformers>=5.0.0 ; python_version < '3.13'"
+    "sentence-transformers>=5.0.0 ; python_version < '3.13'",
+    "torch>=2.6.0 ; python_version < '3.13'"
 ]
-pinecone = ["pinecone>=5.0.0,<6.0.0"]
 vision = [
     "torchvision>=0.17.0 ; python_version < '3.13'",
     "transformers>=4.36.2 ; python_version < '3.13'",
     "pillow>=10.2.0,<11.0.0 ; python_version < '3.13'",
+    "torch>=2.6.0 ; python_version < '3.13'"
 ]
+pinecone = ["pinecone>=5.0.0,<6.0.0"]
 mistralai = ["mistralai>=0.0.12,<0.1.0"]
 qdrant = ["qdrant-client>=1.11.1,<2"]
 google = ["google-cloud-aiplatform>=1.45.0,<2"]

semantic_router/index/hybrid_local.py

@@ -16,6 +16,7 @@ class HybridLocalIndex(LocalIndex):
     def __init__(self, **data):
         super().__init__(**data)
         self.metadata = None
+        
 
     def add(
         self,