Added test for AlbertForTokenClassification

f7115752 · Martin Malmsten · aceb6a09 · f7115752
Commit f7115752 authored Feb 27, 2020 by Martin Malmsten
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/test_modeling_albert.py tests/test_modeling_albert.py +1 -0

No files found.
--- a/tests/test_modeling_albert.py
+++ b/tests/test_modeling_albert.py
@@ -29,6 +29,7 @@ if is_torch_available():
        AlbertModel,
        AlbertForMaskedLM,
        AlbertForSequenceClassification,
+        AlbertForTokenClassification,
        AlbertForQuestionAnswering,
    )
    from transformers.modeling_albert import ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP