fix test

kderusso · May 3, 2024 · c0ec664 · c0ec664
1 parent 4b30b6a
commit c0ec664
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/ml/search/WeightedToken.java b/x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/ml/search/WeightedToken.java
@@ -20,6 +20,7 @@
 
 import java.io.IOException;
 import java.util.Map;
+import java.util.Objects;
 
 public class WeightedToken implements Writeable, ToXContentFragment {
 
@@ -76,4 +77,17 @@ public static WeightedToken fromXContent(XContentParser parser) {
             throw new ParsingException(parser.getTokenLocation(), e.getMessage(), e);
         }
     }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        WeightedToken that = (WeightedToken) o;
+        return Float.compare(weight, that.weight) == 0 && Objects.equals(token, that.token);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(token, weight);
+    }
 }