Keep features in the order they are in when we receive them in a proto

stanfordnlp · Mar 11, 2023 · 9ea4f39 · 9ea4f39
1 parent 901b0f0
commit 9ea4f39
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/edu/stanford/nlp/pipeline/ProtobufAnnotationSerializer.java b/src/edu/stanford/nlp/pipeline/ProtobufAnnotationSerializer.java
@@ -2551,7 +2551,9 @@ public static SentenceFragment fromProto(CoreNLPProtos.SentenceFragment fragment
    * @return A Java Map corresponding to the serialized map.
    */
   public static HashMap<String, String> fromProto(CoreNLPProtos.MapStringString proto) {
-    HashMap<String, String> map = new HashMap<>();
+    // LinkedHashMap so that the pieces stay in the order we receive them
+    // (could also use a TreeMap instead to enforce sorted by key)
+    HashMap<String, String> map = new LinkedHashMap<>();
     for (int i = 0; i < proto.getKeyCount(); ++i) {
       map.put(proto.getKey(i), proto.getValue(i));
     }