Skip to content

Commit c8647ec

Browse files
committed
Update embedders distribution test
1 parent 10a5eaa commit c8647ec

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/test/java/com/meilisearch/integration/SettingsTest.java

+4-2
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import com.meilisearch.sdk.Index;
1919
import com.meilisearch.sdk.model.EmbedderSource;
2020
import com.meilisearch.sdk.model.Embedder;
21+
import com.meilisearch.sdk.model.EmbedderDistribution;
2122
import com.meilisearch.sdk.model.FacetSortValue;
2223
import com.meilisearch.sdk.model.Faceting;
2324
import com.meilisearch.sdk.model.LocalizedAttribute;
@@ -1504,7 +1505,7 @@ public void testUpdateEmbeddersSettings() throws Exception {
15041505
.setSource(EmbedderSource.HUGGING_FACE)
15051506
.setModel("sentence-transformers/all-MiniLM-L6-v2")
15061507
.setRevision("main")
1507-
.setDistribution("uniform");
1508+
.setDistribution(EmbedderDistribution.uniform());
15081509

15091510
// Test REST embedder with request and response
15101511
Map<String, Object> request = new HashMap<>();
@@ -1569,7 +1570,8 @@ public void testUpdateEmbeddersSettings() throws Exception {
15691570
assertThat(retrievedHf.getSource(), is(equalTo(EmbedderSource.HUGGING_FACE)));
15701571
assertThat(retrievedHf.getModel(), is(equalTo("sentence-transformers/all-MiniLM-L6-v2")));
15711572
assertThat(retrievedHf.getRevision(), is(equalTo("main")));
1572-
assertThat(retrievedHf.getDistribution(), is(equalTo("uniform")));
1573+
assertThat(retrievedHf.getDistribution().getMean(), is(equalTo(0.5)));
1574+
assertThat(retrievedHf.getDistribution().getSigma(), is(equalTo(0.5)));
15731575

15741576
// Check REST embedder
15751577
Embedder retrievedRest = updatedEmbedders.get("rest");

0 commit comments

Comments
 (0)