riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Stas Bekman 2020-07-13 03:09:49 -07:00 committed by GitHub
parent 74843695eb
commit 443b0cad96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_modeling_xlm.py
View File

@ -297,7 +297,7 @@ class XLMModelTester:
self.parent.assertListEqual(list(result["loss"].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,
config,
input_ids,
@ -383,9 +383,9 @@ class XLMModelTest(ModelTesterMixin, unittest.TestCase):
config_and_inputs = self.model_tester.prepare_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()
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
def test_model_from_pretrained(self):