Skip to content

Commit

Permalink
spotlessApply styling
Browse files Browse the repository at this point in the history
Signed-off-by: Sahil Buddharaju <[email protected]>
  • Loading branch information
Sahil Buddharaju committed Nov 26, 2024
1 parent 4ac28af commit bae4573
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 39 deletions.
48 changes: 24 additions & 24 deletions src/test/java/org/opensearch/knn/integ/ModeAndCompressionIT.java
Original file line number Diff line number Diff line change
Expand Up @@ -268,30 +268,30 @@ public void testTraining_whenInvalid_thenFail() {
String modelId = "test";

XContentBuilder builder1 = XContentFactory.jsonBuilder()
.startObject()
.field(TRAIN_INDEX_PARAMETER, TRAINING_INDEX_NAME)
.field(TRAIN_FIELD_PARAMETER, TRAINING_FIELD_NAME)
.field(KNNConstants.DIMENSION, DIMENSION)
.field(VECTOR_DATA_TYPE_FIELD, "float")
.field(MODEL_DESCRIPTION, "")
.field(MODE_PARAMETER, Mode.ON_DISK)
.field(COMPRESSION_LEVEL_PARAMETER, "16x")
.startObject(KNN_METHOD)
.field(NAME, METHOD_IVF)
.field(KNN_ENGINE, FAISS_NAME)
.field(METHOD_PARAMETER_SPACE_TYPE, "l2")
.startObject(PARAMETERS)
.field(METHOD_PARAMETER_NLIST, 1)
.startObject(METHOD_ENCODER_PARAMETER)
.field(NAME, "pq")
.startObject(PARAMETERS)
.field(ENCODER_PARAMETER_PQ_CODE_SIZE, 2)
.field(ENCODER_PARAMETER_PQ_M, 8)
.endObject()
.endObject()
.endObject()
.endObject()
.endObject();
.startObject()
.field(TRAIN_INDEX_PARAMETER, TRAINING_INDEX_NAME)
.field(TRAIN_FIELD_PARAMETER, TRAINING_FIELD_NAME)
.field(KNNConstants.DIMENSION, DIMENSION)
.field(VECTOR_DATA_TYPE_FIELD, "float")
.field(MODEL_DESCRIPTION, "")
.field(MODE_PARAMETER, Mode.ON_DISK)
.field(COMPRESSION_LEVEL_PARAMETER, "16x")
.startObject(KNN_METHOD)
.field(NAME, METHOD_IVF)
.field(KNN_ENGINE, FAISS_NAME)
.field(METHOD_PARAMETER_SPACE_TYPE, "l2")
.startObject(PARAMETERS)
.field(METHOD_PARAMETER_NLIST, 1)
.startObject(METHOD_ENCODER_PARAMETER)
.field(NAME, "pq")
.startObject(PARAMETERS)
.field(ENCODER_PARAMETER_PQ_CODE_SIZE, 2)
.field(ENCODER_PARAMETER_PQ_M, 8)
.endObject()
.endObject()
.endObject()
.endObject()
.endObject();
expectThrows(ResponseException.class, () -> trainModel(modelId, builder1));

XContentBuilder builder2 = XContentFactory.jsonBuilder()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -515,24 +515,24 @@ public void testTrainModel_success_methodOverrideWithCompressionMode() throws Ex
*/
XContentBuilder builder = XContentFactory.jsonBuilder()
.startObject()
.field(NAME, "ivf")
.startObject(PARAMETERS)
.field(METHOD_PARAMETER_NLIST, 16)
.endObject()
.endObject();
.startObject()
.field(NAME, "ivf")
.startObject(PARAMETERS)
.field(METHOD_PARAMETER_NLIST, 16)
.endObject()
.endObject();
Map<String, Object> method = xContentBuilderToMap(builder);

XContentBuilder outerParams = XContentFactory.jsonBuilder()
.startObject()
.field(TRAIN_INDEX_PARAMETER, trainingIndexName)
.field(TRAIN_FIELD_PARAMETER, nestedFieldPath)
.field(DIMENSION, dimension)
.field(COMPRESSION_LEVEL_PARAMETER, "16x")
.field(MODE_PARAMETER, "on_disk")
.field(KNN_METHOD, method)
.field(MODEL_DESCRIPTION, "dummy description")
.endObject();
.startObject()
.field(TRAIN_INDEX_PARAMETER, trainingIndexName)
.field(TRAIN_FIELD_PARAMETER, nestedFieldPath)
.field(DIMENSION, dimension)
.field(COMPRESSION_LEVEL_PARAMETER, "16x")
.field(MODE_PARAMETER, "on_disk")
.field(KNN_METHOD, method)
.field(MODEL_DESCRIPTION, "dummy description")
.endObject();

Request request = new Request("POST", "/_plugins/_knn/models/" + modelId + "/_train");
request.setJsonEntity(outerParams.toString());
Expand Down

0 comments on commit bae4573

Please sign in to comment.