def METHOD_NAME(dataset_token_classification) -> None:
    trainer = ArgillaTrainer(name=dataset_token_classification, model=MODEL, framework=FRAMEWORK)
    record = trainer.predict("This is a text", as_argilla_records=True)
    assert isinstance(record, TokenClassificationRecord)
    not_record = trainer.predict("This is a text", as_argilla_records=False)
    assert not isinstance(not_record, TokenClassificationRecord)
    record_list = trainer.predict(["This is a text"], as_argilla_records=True)
    assert isinstance(record_list, list)
    assert len(record_list) == 1
    assert isinstance(record_list[0], TokenClassificationRecord)
    not_record_list = trainer.predict(["This is a text"], as_argilla_records=False)
    assert isinstance(not_record_list, list)
    assert len(not_record_list) == 1
    assert not isinstance(not_record_list[0], TokenClassificationRecord)