def test_sequence_classification_explainer_str(): seq_explainer = SequenceClassificationExplainer(DISTILBERT_MODEL, DISTILBERT_TOKENIZER) s = "SequenceClassificationExplainer(" s += f"\n\tmodel={DISTILBERT_MODEL.__class__.__name__}," s += f"\n\ttokenizer={DISTILBERT_TOKENIZER.__class__.__name__}," s += "\n\tattribution_type='lig'," s += ")" assert s == seq_explainer.__str__()
def test_sequence_classification_explainer_str(): explainer_string = "I love you , I like you" seq_explainer = SequenceClassificationExplainer(explainer_string, MODEL, TOKENIZER) s = "SequenceClassificationExplainer(" s += f'\n\ttext="{explainer_string[:10]}...",' s += f"\n\tmodel={MODEL.__class__.__name__}," s += f"\n\ttokenizer={TOKENIZER.__class__.__name__}," s += "\n\tattribution_type='lig'," s += ")" assert s == seq_explainer.__str__()