From e8256c49dc73c0ac93dff427bbeb3bcbfde5dd8b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dominique=20Eifl=C3=A4nder?= Date: Tue, 5 Jan 2021 14:53:09 +0100 Subject: [PATCH] Fixed annotating cell with more than one TextBlock (Mismatch between EntityPositionSequence and found Entity) --- .../redaction/v1/server/redaction/model/Section.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java index 3e534656..0d6d497b 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/model/Section.java @@ -2,15 +2,18 @@ package com.iqser.red.service.redaction.v1.server.redaction.model; import static com.iqser.red.service.redaction.v1.server.redaction.model.Dictionary.RECOMMENDATION_PREFIX; +import java.util.Collection; import java.util.HashMap; import java.util.HashSet; import java.util.Map; import java.util.Set; import java.util.regex.Matcher; import java.util.regex.Pattern; +import java.util.stream.Collectors; import org.apache.commons.lang3.StringUtils; +import com.iqser.red.service.redaction.v1.server.classification.model.TextBlock; import com.iqser.red.service.redaction.v1.server.redaction.utils.Patterns; import com.iqser.red.service.redaction.v1.server.redaction.utils.PositionUtil; @@ -292,8 +295,10 @@ public class Section { entity.setMatchedRule(ruleNumber); entity.setRedactionReason(reason); entity.setTargetSequences(value.getTextBlocks() - .get(0) - .getSequences()); // Make sure no other cells with same content are highlighted + .stream() + .map(TextBlock::getSequences) + .flatMap(Collection::stream) + .collect(Collectors.toList())); // Make sure no other cells with same content are highlighted entity.setLegalBasis(legalBasis); Set singleEntitySet = new HashSet<>();