src/adapters/ml/fast_trainer/Paragraph.py

from pdf_features import PdfToken


class Paragraph:
    def __init__(self, tokens: list[PdfToken], pdf_name: str = ""):
        self.tokens = tokens
        self.pdf_name = pdf_name

    def add_token(self, token: PdfToken):
        self.tokens.append(token)