mirror of
https://github.com/elastic/eland.git
synced 2025-07-24 00:00:39 +08:00
Compare commits
No commits in common. "main" and "v9.0.1" have entirely different histories.
2
.github/workflows/docs-build.yml
vendored
2
.github/workflows/docs-build.yml
vendored
@ -16,4 +16,4 @@ jobs:
|
||||
deployments: write
|
||||
id-token: write
|
||||
contents: read
|
||||
pull-requests: write
|
||||
pull-requests: read
|
||||
|
@ -53,8 +53,7 @@ $ conda install -c conda-forge eland
|
||||
|
||||
### Compatibility
|
||||
|
||||
- Supports Python 3.9, 3.10, 3.11 and 3.12.
|
||||
- Supports Pandas 1.5 and 2.
|
||||
- Supports Python 3.9, 3.10, 3.11, 3.12 and Pandas 1.5
|
||||
- Supports Elasticsearch 8+ clusters, recommended 8.16 or later for all features to work.
|
||||
If you are using the NLP with PyTorch feature make sure your Eland minor version matches the minor
|
||||
version of your Elasticsearch cluster. For all other features it is sufficient for the major versions
|
||||
|
@ -1,6 +1,4 @@
|
||||
project: 'Eland Python client'
|
||||
products:
|
||||
- id: elasticsearch-client
|
||||
cross_links:
|
||||
- docs-content
|
||||
toc:
|
||||
|
@ -50,7 +50,10 @@ class Index:
|
||||
# index_field.setter
|
||||
self._is_source_field = False
|
||||
|
||||
self.es_index_field = es_index_field
|
||||
# The type:ignore is due to mypy not being smart enough
|
||||
# to recognize the property.setter has a different type
|
||||
# than the property.getter.
|
||||
self.es_index_field = es_index_field # type: ignore
|
||||
|
||||
@property
|
||||
def sort_field(self) -> str:
|
||||
|
4
setup.py
4
setup.py
@ -62,10 +62,10 @@ extras = {
|
||||
"requests<3",
|
||||
"torch==2.5.1",
|
||||
"tqdm",
|
||||
"sentence-transformers>=5.0.0,<6.0.0",
|
||||
"sentence-transformers>=2.1.0,<=2.7.0",
|
||||
# sentencepiece is a required dependency for the slow tokenizers
|
||||
# https://huggingface.co/transformers/v4.4.2/migration.html#sentencepiece-is-removed-from-the-required-dependencies
|
||||
"transformers[sentencepiece]>=4.47.0,<4.50.3",
|
||||
"transformers[sentencepiece]>=4.47.0",
|
||||
],
|
||||
}
|
||||
extras["all"] = list({dep for deps in extras.values() for dep in deps})
|
||||
|
@ -65,8 +65,6 @@ TEXT_EMBEDDING_MODELS = [
|
||||
|
||||
TEXT_SIMILARITY_MODELS = ["mixedbread-ai/mxbai-rerank-xsmall-v1"]
|
||||
|
||||
TEXT_EXPANSION_MODELS = ["naver/splade-v3-distilbert"]
|
||||
|
||||
|
||||
@pytest.fixture(scope="function", autouse=True)
|
||||
def setup_and_tear_down():
|
||||
@ -157,22 +155,3 @@ class TestPytorchModel:
|
||||
|
||||
assert result.body["inference_results"][0]["predicted_value"] < 0
|
||||
assert result.body["inference_results"][1]["predicted_value"] > 0
|
||||
|
||||
@pytest.mark.skipif(ES_VERSION < (9, 0, 0), reason="requires current major version")
|
||||
@pytest.mark.parametrize("model_id", TEXT_EXPANSION_MODELS)
|
||||
def test_text_expansion(self, model_id):
|
||||
with tempfile.TemporaryDirectory() as tmp_dir:
|
||||
ptm = download_model_and_start_deployment(
|
||||
tmp_dir, False, model_id, "text_expansion"
|
||||
)
|
||||
result = ptm.infer(
|
||||
docs=[
|
||||
{
|
||||
"text_field": "The Amazon rainforest covers most of the Amazon basin in South America"
|
||||
},
|
||||
{"text_field": "Paris is the capital of France"},
|
||||
]
|
||||
)
|
||||
|
||||
assert len(result.body["inference_results"][0]["predicted_value"]) > 0
|
||||
assert len(result.body["inference_results"][1]["predicted_value"]) > 0
|
||||
|
Loading…
x
Reference in New Issue
Block a user