rename the function to match the rest of the test convention (#5692)
This commit is contained in:
Родитель
74843695eb
Коммит
443b0cad96
|
@ -297,7 +297,7 @@ class XLMModelTester:
|
||||||
self.parent.assertListEqual(list(result["loss"].size()), [])
|
self.parent.assertListEqual(list(result["loss"].size()), [])
|
||||||
self.parent.assertListEqual(list(result["logits"].size()), [self.batch_size, self.type_sequence_label_size])
|
self.parent.assertListEqual(list(result["logits"].size()), [self.batch_size, self.type_sequence_label_size])
|
||||||
|
|
||||||
def create_and_check_xlm_for_token_classification(
|
def create_and_check_xlm_token_classif(
|
||||||
self,
|
self,
|
||||||
config,
|
config,
|
||||||
input_ids,
|
input_ids,
|
||||||
|
@ -383,9 +383,9 @@ class XLMModelTest(ModelTesterMixin, unittest.TestCase):
|
||||||
config_and_inputs = self.model_tester.prepare_config_and_inputs()
|
config_and_inputs = self.model_tester.prepare_config_and_inputs()
|
||||||
self.model_tester.create_and_check_xlm_sequence_classif(*config_and_inputs)
|
self.model_tester.create_and_check_xlm_sequence_classif(*config_and_inputs)
|
||||||
|
|
||||||
def test_xlm_for_token_classification(self):
|
def test_xlm_token_classif(self):
|
||||||
config_and_inputs = self.model_tester.prepare_config_and_inputs()
|
config_and_inputs = self.model_tester.prepare_config_and_inputs()
|
||||||
self.model_tester.create_and_check_xlm_for_token_classification(*config_and_inputs)
|
self.model_tester.create_and_check_xlm_token_classif(*config_and_inputs)
|
||||||
|
|
||||||
@slow
|
@slow
|
||||||
def test_model_from_pretrained(self):
|
def test_model_from_pretrained(self):
|
||||||
|
|
Загрузка…
Ссылка в новой задаче