Skip to content

Commit

Permalink
Fix test model name (#2538) (#2544)
Browse files Browse the repository at this point in the history
(cherry picked from commit 6fd8afa)

Co-authored-by: Max Jakob <max.jakob@elastic.co>
  • Loading branch information
github-actions[bot] and maxjakob committed May 2, 2024
1 parent 590ee66 commit fd699d5
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,15 @@ services:
retries: 60

# Currently fails on Mac: https://github.com/elastic/elasticsearch/issues/106206
elasticsearch-with-models:
elasticsearch-with-model:
image: docker.elastic.co/eland/eland
depends_on:
- elasticsearch
restart: no
command: sh -c "sleep 10 && eland_import_hub_model --url http://elasticsearch:9200 --hub-model-id sentence-transformers/msmarco-minilm-l-12-v3 --start"
command: sh -c "
sleep 10 &&
eland_import_hub_model \
--hub-model-id sentence-transformers/all-minilm-l6-v2 \
--url http://elasticsearch:9200 \
--start
"
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@

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


Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -525,9 +525,7 @@ def test_search_with_knn_infer_instack(

store = VectorStore(
index=index,
retrieval_strategy=DenseVectorStrategy(
model_id="sentence-transformers__all-minilm-l6-v2"
),
retrieval_strategy=DenseVectorStrategy(model_id=TRANSFORMER_MODEL_ID),
client=sync_client,
)

Expand Down

0 comments on commit fd699d5

Please sign in to comment.