diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/BUILD b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/BUILD
deleted file mode 100644
index dabf212d7e7162849c24a733909ac4f645d75a31..0000000000000000000000000000000000000000
--- a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/BUILD
+++ /dev/null
@@ -1 +0,0 @@
-python_tests()
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/__init__.py b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/__init__.py
deleted file mode 100644
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/test_embeddings_huggingface_itrex.py b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/test_embeddings_huggingface_itrex.py
deleted file mode 100644
index 4bf279207529450acb0607c1ab105459daf77c36..0000000000000000000000000000000000000000
--- a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/test_embeddings_huggingface_itrex.py
+++ /dev/null
@@ -1,7 +0,0 @@
-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
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-optimum/tests/test_embeddings_huggingface_optimum.py b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-optimum/tests/test_embeddings_huggingface_optimum.py
index 5c1b57df5a6380b76334d4aa72a2e5b53264036e..6b48f781928e4cf23006329eb4dc4f9cb829ab7a 100644
--- a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-optimum/tests/test_embeddings_huggingface_optimum.py
+++ b/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
diff --git a/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/BUILD b/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/BUILD
deleted file mode 100644
index dabf212d7e7162849c24a733909ac4f645d75a31..0000000000000000000000000000000000000000
--- a/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/BUILD
+++ /dev/null
@@ -1 +0,0 @@
-python_tests()
diff --git a/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/__init__.py b/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/__init__.py
deleted file mode 100644
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000
diff --git a/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/test_llms_nvidia_triton.py b/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/test_llms_nvidia_triton.py
deleted file mode 100644
index c0058b15a1ce159fc558888e4a5f0bafcdc184aa..0000000000000000000000000000000000000000
--- a/llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/test_llms_nvidia_triton.py
+++ /dev/null
@@ -1,7 +0,0 @@
-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
diff --git a/pants.toml b/pants.toml
index 79bfbb34c0d08d6567d70888cce66e3444219ed3..fd2d071fcf9ef88e9f7d8374e8874954679e4e5b 100644
--- a/pants.toml
+++ b/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]