def make_digest(entity_name: str, arxiv_id: ArxivId) -> EntityProcessingDigest:
default_digest = make_default_paper_digest(entity_name, arxiv_id)
return EntityProcessingDigest(
num_extracted=count_entities_extracted(arxiv_id),
num_hues_located=default_digest.num_hues_located,
num_entities_located=count_entities_located(arxiv_id),
)
def make_digest(_: str, arxiv_id: ArxivId) -> EntityProcessingDigest:
"""
Custom digest creator. Count the equation tokens, instead of the 'symbols', as we can
use the default entity counters for the outputs of equation token commands.
"""
return make_default_paper_digest("equation-tokens", arxiv_id)