Skip to content

Commit e146cda

Browse files
committed
Unmute SemanticInferenceMetadataFieldsRecoveryTests#testSnapshotRecovery
Modify the test to avoid using cosine similarity, as random vectors may fail the magnitude check. Closes #122549 Closes #122550 Closes #122551
1 parent 731a412 commit e146cda

File tree

2 files changed

+4
-13
lines changed

2 files changed

+4
-13
lines changed

muted-tests.yml

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -271,15 +271,6 @@ tests:
271271
- class: org.elasticsearch.smoketest.DocsClientYamlTestSuiteIT
272272
method: test {yaml=reference/snapshot-restore/apis/get-snapshot-api/line_488}
273273
issue: https://github.com/elastic/elasticsearch/issues/121611
274-
- class: org.elasticsearch.xpack.inference.mapper.SemanticInferenceMetadataFieldsRecoveryTests
275-
method: testSnapshotRecovery {p0=false p1=false}
276-
issue: https://github.com/elastic/elasticsearch/issues/122549
277-
- class: org.elasticsearch.xpack.inference.mapper.SemanticInferenceMetadataFieldsRecoveryTests
278-
method: testSnapshotRecovery {p0=true p1=false}
279-
issue: https://github.com/elastic/elasticsearch/issues/122550
280-
- class: org.elasticsearch.xpack.inference.mapper.SemanticInferenceMetadataFieldsRecoveryTests
281-
method: testSnapshotRecovery {p0=false p1=true}
282-
issue: https://github.com/elastic/elasticsearch/issues/122551
283274
- class: org.elasticsearch.index.mapper.ShapeGeometryFieldMapperTests
284275
method: testCartesianBoundsBlockLoader
285276
issue: https://github.com/elastic/elasticsearch/issues/122661

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/mapper/SemanticInferenceMetadataFieldsRecoveryTests.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,7 @@ protected List<MapperPlugin> extraMappers() {
7373

7474
@Override
7575
protected Settings indexSettings() {
76-
var builder = Settings.builder()
77-
.put(super.indexSettings())
78-
.put(InferenceMetadataFieldsMapper.USE_LEGACY_SEMANTIC_TEXT_FORMAT.getKey(), false);
76+
var builder = Settings.builder().put(super.indexSettings());
7977
if (useSynthetic) {
8078
builder.put(IndexSettings.INDEX_MAPPER_SOURCE_MODE_SETTING.getKey(), SourceFieldMapper.Mode.SYNTHETIC.name());
8179
builder.put(IndexSettings.RECOVERY_USE_SYNTHETIC_SOURCE_SETTING.getKey(), true);
@@ -222,7 +220,9 @@ private Translog.Snapshot newRandomSnapshot(
222220

223221
private static Model randomModel(TaskType taskType) {
224222
var dimensions = taskType == TaskType.TEXT_EMBEDDING ? randomIntBetween(2, 64) : null;
225-
var similarity = taskType == TaskType.TEXT_EMBEDDING ? randomFrom(SimilarityMeasure.values()) : null;
223+
var similarity = taskType == TaskType.TEXT_EMBEDDING
224+
? randomValueOtherThan(SimilarityMeasure.COSINE, () -> randomFrom(SimilarityMeasure.values()))
225+
: null;
226226
var elementType = taskType == TaskType.TEXT_EMBEDDING ? DenseVectorFieldMapper.ElementType.BYTE : null;
227227
return new TestModel(
228228
randomAlphaOfLength(4),

0 commit comments

Comments
 (0)