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__()
