fix wrong label adjustment

kermitt2 · Dec 6, 2024 · 17ad425 · 17ad425
1 parent facc35e
commit 17ad425
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/grobid-core/src/main/java/org/grobid/core/utilities/LabelUtils.java b/grobid-core/src/main/java/org/grobid/core/utilities/LabelUtils.java
@@ -56,15 +56,11 @@ public static String adjustInvalidSequenceOfStartLabels(String fulltextLabeledTe
             String[] pieces = line.split("\t");
             String label = pieces[pieces.length-1];
             if (label.equals("I-"+TaggingLabels.FIGURE.getLabel())) {
-                if (previousLabel == null) {
-                    continue;
-                } else if (previousLabel.equals("I-"+TaggingLabels.FIGURE.getLabel())) {
+                if (StringUtils.equals(previousLabel, "I-"+TaggingLabels.FIGURE.getLabel())) {
                     pieces[pieces.length-1] = TaggingLabels.FIGURE.getLabel();
                 }
             } else if (label.equals("I-"+TaggingLabels.TABLE.getLabel())) {
-                if (previousLabel == null) {
-                    continue;
-                } else if (previousLabel.equals("I-"+TaggingLabels.TABLE.getLabel())) {
+                if (StringUtils.equals(previousLabel, "I-"+TaggingLabels.TABLE.getLabel())) {
                     pieces[pieces.length-1] = TaggingLabels.TABLE.getLabel();
                 }
             }