From db36fd29f2f31996f6a168e4451744ade64e8c28 Mon Sep 17 00:00:00 2001
From: Logan <logan.markewich@live.com>
Date: Wed, 3 Apr 2024 16:54:43 -0600
Subject: [PATCH] pants fixes (#12559)
---
.../llama-index-embeddings-huggingface-itrex/tests/BUILD | 1 -
.../tests/__init__.py | 0
.../tests/test_embeddings_huggingface_itrex.py | 7 -------
.../tests/test_embeddings_huggingface_optimum.py | 5 ++++-
.../llms/llama-index-llms-nvidia-triton/tests/BUILD | 1 -
.../llms/llama-index-llms-nvidia-triton/tests/__init__.py | 0
.../tests/test_llms_nvidia_triton.py | 7 -------
pants.toml | 3 +++
8 files changed, 7 insertions(+), 17 deletions(-)
delete mode 100644 llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/BUILD
delete mode 100644 llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/__init__.py
delete mode 100644 llama-index-integrations/embeddings/llama-index-embeddings-huggingface-itrex/tests/test_embeddings_huggingface_itrex.py
delete mode 100644 llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/BUILD
delete mode 100644 llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/__init__.py
delete mode 100644 llama-index-integrations/llms/llama-index-llms-nvidia-triton/tests/test_llms_nvidia_triton.py
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 dabf212d7..000000000
--- 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 e69de29bb..000000000
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 4bf279207..000000000
--- 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 5c1b57df5..6b48f7819 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 dabf212d7..000000000
--- 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 e69de29bb..000000000
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 c0058b15a..000000000
--- 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 79bfbb34c..fd2d071fc 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]
--
GitLab