From f71157529e865f0195f226c9a5648d9a3f3e0f41 Mon Sep 17 00:00:00 2001
From: Martin Malmsten <martin.malmsten@kb.se>
Date: Thu, 27 Feb 2020 12:24:20 +0100
Subject: [PATCH] Added test for AlbertForTokenClassification

---
 tests/test_modeling_albert.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_modeling_albert.py b/tests/test_modeling_albert.py
index 389887c47c..b1fe4bb64e 100644
--- 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