"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a0906068cfe337404f2e2f4d0328b04fa15945cf"
Unverified Commit 443b0cad authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

rename the function to match the rest of the test convention (#5692)

parent 74843695
...@@ -297,7 +297,7 @@ class XLMModelTester: ...@@ -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): ...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment