Skip to content

Commit 85207ae

Browse files
Fix test model name (#2538) (#2543)
(cherry picked from commit 6fd8afa) Co-authored-by: Max Jakob <max.jakob@elastic.co>
1 parent 3074c85 commit 85207ae

File tree

3 files changed

+10
-6
lines changed

3 files changed

+10
-6
lines changed

test_elasticsearch/test_server/test_vectorstore/docker-compose.yml

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,15 @@ services:
2020
retries: 60
2121

2222
# Currently fails on Mac: https://github.com/elastic/elasticsearch/issues/106206
23-
elasticsearch-with-models:
23+
elasticsearch-with-model:
2424
image: docker.elastic.co/eland/eland
2525
depends_on:
2626
- elasticsearch
2727
restart: no
28-
command: sh -c "sleep 10 && eland_import_hub_model --url http://elasticsearch:9200 --hub-model-id sentence-transformers/msmarco-minilm-l-12-v3 --start"
28+
command: sh -c "
29+
sleep 10 &&
30+
eland_import_hub_model \
31+
--hub-model-id sentence-transformers/all-minilm-l6-v2 \
32+
--url http://elasticsearch:9200 \
33+
--start
34+
"

test_elasticsearch/test_server/test_vectorstore/test_embedding_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626

2727
# deployed with
2828
# https://www.elastic.co/guide/en/machine-learning/current/ml-nlp-text-emb-vector-search-example.html
29-
MODEL_ID = os.getenv("MODEL_ID", "sentence-transformers__msmarco-minilm-l-12-v3")
29+
MODEL_ID = os.getenv("MODEL_ID", "sentence-transformers__all-minilm-l6-v2")
3030
NUM_DIMENSIONS = int(os.getenv("NUM_DIMENSIONS", "384"))
3131

3232

test_elasticsearch/test_server/test_vectorstore/test_vectorstore.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -525,9 +525,7 @@ def test_search_with_knn_infer_instack(
525525

526526
store = VectorStore(
527527
index=index,
528-
retrieval_strategy=DenseVectorStrategy(
529-
model_id="sentence-transformers__all-minilm-l6-v2"
530-
),
528+
retrieval_strategy=DenseVectorStrategy(model_id=TRANSFORMER_MODEL_ID),
531529
client=sync_client,
532530
)
533531

0 commit comments

Comments
 (0)