From 327833a77f1dd99ad20b7853177a2338ff2c35cd Mon Sep 17 00:00:00 2001 From: xuyige Date: Fri, 12 Jul 2019 04:11:03 +0800 Subject: [PATCH] fix bug in test code --- test/models/test_bert.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/models/test_bert.py b/test/models/test_bert.py index 38a16f9b..05ee6d5a 100644 --- a/test/models/test_bert.py +++ b/test/models/test_bert.py @@ -8,7 +8,7 @@ from fastNLP.models.bert import * class TestBert(unittest.TestCase): def test_bert_1(self): from fastNLP.core.const import Const - from fastNLP.modules.encoder._bert import BertConfig + from fastNLP.modules.encoder.bert import BertConfig model = BertForSequenceClassification(2, BertConfig(32000)) @@ -23,7 +23,7 @@ class TestBert(unittest.TestCase): def test_bert_2(self): from fastNLP.core.const import Const - from fastNLP.modules.encoder._bert import BertConfig + from fastNLP.modules.encoder.bert import BertConfig model = BertForMultipleChoice(2, BertConfig(32000)) @@ -38,7 +38,7 @@ class TestBert(unittest.TestCase): def test_bert_3(self): from fastNLP.core.const import Const - from fastNLP.modules.encoder._bert import BertConfig + from fastNLP.modules.encoder.bert import BertConfig model = BertForTokenClassification(7, BertConfig(32000)) @@ -53,7 +53,7 @@ class TestBert(unittest.TestCase): def test_bert_4(self): from fastNLP.core.const import Const - from fastNLP.modules.encoder._bert import BertConfig + from fastNLP.modules.encoder.bert import BertConfig model = BertForQuestionAnswering(BertConfig(32000))