From 2fdc53429ca990d7900a27f36f08c7381bf9df21 Mon Sep 17 00:00:00 2001 From: Kilian Schuettler Date: Thu, 28 Nov 2024 12:57:13 +0100 Subject: [PATCH] fix accidental push to main --- .../processor/services/factory/DocumentGraphFactory.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/DocumentGraphFactory.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/DocumentGraphFactory.java index 97bad8e..7ecfd6c 100644 --- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/DocumentGraphFactory.java +++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/DocumentGraphFactory.java @@ -16,7 +16,7 @@ import java.util.Optional; import java.util.Set; import java.util.stream.Collectors; -import com.knecon.fforesight.service.layoutparser.internal.api.data.redaction.NodeTypeProto; +import com.knecon.fforesight.service.layoutparser.internal.api.data.redaction.NodeType; import com.knecon.fforesight.service.layoutparser.internal.api.queue.LayoutParsingType; import com.knecon.fforesight.service.layoutparser.processor.model.AbstractPageBlock; import com.knecon.fforesight.service.layoutparser.processor.model.ClassificationDocument; @@ -82,9 +82,9 @@ public class DocumentGraphFactory { documentGraph.streamAllSubNodes() .filter(SemanticNode::isLeaf) - .filter(node -> !node.getType().equals(NodeTypeProto.NodeType.HEADER)) - .filter(node -> !node.getType().equals(NodeTypeProto.NodeType.FOOTER)) - .filter(node -> !node.getType().equals(NodeTypeProto.NodeType.IMAGE)) + .filter(node -> !node.getType().equals(NodeType.HEADER)) + .filter(node -> !node.getType().equals(NodeType.FOOTER)) + .filter(node -> !node.getType().equals(NodeType.IMAGE)) .map(SemanticNode::getTextBlock) .map(TextBlock::getAtomicTextBlocks) .flatMap(Collection::stream)