pre commit

davidkyle · Nov 22, 2023 · dfdd495 · dfdd495
1 parent a4b3cc3
commit dfdd495
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/...tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/MockInferenceServiceIT.java b/...tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/MockInferenceServiceIT.java
@@ -14,7 +14,6 @@
 import org.elasticsearch.common.settings.SecureString;
 import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.common.util.concurrent.ThreadContext;
-import org.elasticsearch.inference.ModelConfigurations;
 import org.elasticsearch.inference.TaskType;
 import org.elasticsearch.test.cluster.ElasticsearchCluster;
 import org.elasticsearch.test.cluster.local.distribution.DistributionType;
@@ -97,7 +96,7 @@ public void testMockServiceWithMultipleInputs() throws IOException {
             List.of(randomAlphaOfLength(5), randomAlphaOfLength(10), randomAlphaOfLength(15))
         );
 
-        var tokens = (List<Map<String, Object>>)inference.get("inference_results");
+        var tokens = (List<Map<String, Object>>) inference.get("inference_results");
         assertThat(tokens, hasSize(3));
         assertNonEmptyInferenceResults(inference, TaskType.SPARSE_EMBEDDING);
     }
@@ -139,11 +138,11 @@ private Map<String, Object> inferOnMockService(String modelId, TaskType taskType
         var request = new Request("POST", endpoint);
 
         var bodyBuilder = new StringBuilder("{\"input\": [");
-        for (var in : input){
+        for (var in : input) {
             bodyBuilder.append('"').append(in).append('"').append(',');
         }
         // remove last comma
-        bodyBuilder.deleteCharAt(bodyBuilder.length() -1);
+        bodyBuilder.deleteCharAt(bodyBuilder.length() - 1);
         bodyBuilder.append("]}");
 
         System.out.println("body_request:" + bodyBuilder);
@@ -156,10 +155,8 @@ private Map<String, Object> inferOnMockService(String modelId, TaskType taskType
     @SuppressWarnings("unchecked")
     protected void assertNonEmptyInferenceResults(Map<String, Object> resultMap, TaskType taskType) {
         if (taskType == TaskType.SPARSE_EMBEDDING) {
-            var tokens = (List<Map<String, Object>>)resultMap.get("inference_results");
-            tokens.forEach(result -> {
-                assertThat(result.keySet(), not(empty()));
-            });
+            var tokens = (List<Map<String, Object>>) resultMap.get("inference_results");
+            tokens.forEach(result -> { assertThat(result.keySet(), not(empty())); });
         } else {
             fail("test with task type [" + taskType + "] are not supported yet");
         }