def METHOD_NAME(self, framework: str, model: str, dataset_text_classification: str) -> None:
    trainer = ArgillaTrainer(name=dataset_text_classification, framework=framework, model=model)
    trainer.update_config(max_steps=1)
    output_dir = "tmp_spacy_train_textcat"
    train_with_cleanup(trainer, output_dir)
    record = trainer.predict("This is a text", as_argilla_records=True)
    assert isinstance(record, rg.TextClassificationRecord)
    assert record.multi_label is False
    not_record = trainer.predict("This is a text", as_argilla_records=False)
    assert isinstance(not_record.cats, dict)
    train_with_cleanup(trainer, output_dir, train=False)
    cleanup_spacy_config(trainer)