Pipeline: bugfix: when colorizing symbols, surround in braces
@@ -4,11 +4,10 @@ from common.commands.base import CommandList
from common.commands.locate_entities import make_locate_entities_command
from common.make_digest import make_default_paper_digest
from common.types import ArxivId, EntityProcessingDigest, SerializableToken
-from scripts.pipelines import EntityPipeline, register_entity_pipeline
-
from entities.sentences.commands.find_entity_sentences import (
make_find_entity_sentences_command,
)
+from scripts.pipelines import EntityPipeline, register_entity_pipeline
from .colorize import adjust_color_positions
from .commands.extract_symbols import ExtractSymbols
@@ -17,7 +16,6 @@ from .commands.find_symbol_sentences import FindSymbolSentences
from .commands.locate_symbols import LocateSymbols
from .commands.upload_symbols import UploadSymbols
-
directories.register("detected-equation-tokens")
directories.register("symbol-matches")
directories.register("sentences-for-equation-tokens")
@@ -38,7 +36,9 @@ commands = [
make_locate_entities_command(
"equation-tokens",
DetectedEntityType=SerializableToken,
- colorize_options=ColorizeOptions(adjust_color_positions=adjust_color_positions),
+ colorize_options=ColorizeOptions(
+ adjust_color_positions=adjust_color_positions, braces=True
+ ),
),
LocateSymbols,
UploadSymbols,
