fix error case with empty affiliation when creating pre-annotated tra…

…ining
kermitt2 · Feb 18, 2024 · d4822e1 · d4822e1
1 parent 3893a8d
commit d4822e1
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/grobid-core/src/main/java/org/grobid/core/engines/FullTextParser.java b/grobid-core/src/main/java/org/grobid/core/engines/FullTextParser.java
@@ -1370,11 +1370,14 @@ public Document createTraining(File inputFile,
                     List<List<LayoutToken>> tokenizationsAffiliation = resHeader.getAffiliationAddresslabeledTokens();
                     //List<LayoutToken> tokenizationsAffiliation = resHeader.getLayoutTokens(TaggingLabels.HEADER_AFFILIATION);
                     List<LayoutToken> tokenizationAffiliation = new ArrayList<>();
-                    for (List<LayoutToken> tokenization : tokenizationsAffiliation) {
-                        tokenizationAffiliation.addAll(tokenization);
+                    StringBuilder bufferAffiliation = null;
+                    if (tokenizationsAffiliation != null && tokenizationsAffiliation.size()>0) {
+                        for (List<LayoutToken> tokenization : tokenizationsAffiliation) {
+                            tokenizationAffiliation.addAll(tokenization);
+                        }
+                        bufferAffiliation =
+                                parsers.getAffiliationAddressParser().trainingExtraction(tokenizationAffiliation);
                     }
-                    StringBuilder bufferAffiliation =
-                            parsers.getAffiliationAddressParser().trainingExtraction(tokenizationAffiliation);
 
                     // buffer for the date block
                     StringBuilder bufferDate = null;