test_CONLL_2003_EL update

TrellixVulnTeam · Jan 20, 2021 · cb39ea7 · cb39ea7
1 parent 07d5335
commit cb39ea7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_entity_vocab/test_CONLL_2003_EL.py b/test/test_entity_vocab/test_CONLL_2003_EL.py
@@ -8,7 +8,7 @@
     BertForPreTraining,
     BertForTokenClassification,
 )
-from hetseq.BertForEL import BertForELClassification
+from hetseq.model.bert_for_EL_classification import BertForELClassification
 
 from transformers import (
     BertTokenizerFast,