pants fixes (#12559)

llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/BUILD

-python_tests()

llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/test_embeddings_huggingface_itrex.py

-from llama_index.core.base.embeddings.base import BaseEmbedding
-from llama_index.embeddings.huggingface_itrex import ItrexQuantizedBgeEmbedding
-
-
-def test_itrex_embedding_class():
-    names_of_base_classes = [b.__name__ for b in ItrexQuantizedBgeEmbedding.__mro__]
-    assert BaseEmbedding.__name__ in names_of_base_classes

llama-index-integrations/embeddings/llama-index-embeddings-huggingface-optimum/tests/test_embeddings_huggingface_optimum.py

@@ -14,10 +14,13 @@ def test_optimumembedding_get_text_embedding(tmp_path):
     embeddings = embed_model.get_text_embedding("Hello World!")
 
     assert len(embeddings) == 384
-    assert embeddings[:5] == [
+    gold_embeddings = [
         -0.0032756966538727283,
         -0.011690770275890827,
         0.04155917093157768,
         -0.038148097693920135,
         0.024183034896850586,
     ]
+
+    for i in range(5):
+        assert abs(embeddings[i] - gold_embeddings[i]) < 1e-4

llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/BUILD

-python_tests()

llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/test_llms_nvidia_triton.py

-from llama_index.core.base.llms.base import BaseLLM
-from llama_index.llms.nvidia_triton import NvidiaTriton
-
-
-def test_embedding_class():
-    names_of_base_classes = [b.__name__ for b in NvidiaTriton.__mro__]
-    assert BaseLLM.__name__ in names_of_base_classes

pants.toml

@@ -7,9 +7,12 @@ backend_packages = [
 ]
 pants_version = "2.19.0"
 remote_cache_read = true
+remote_cache_rpc_timeout_millis = 60000
 remote_cache_write = true
 remote_instance_name = "main"
 remote_store_address = "grpc://framework-ci-balancer-1983728860.us-east-1.elb.amazonaws.com:9092"
+remote_store_rpc_timeout_millis = 60000
+session_end_tasks_timeout = 60.0
 unmatched_build_file_globs = "error"
 
 [black]