From 443b0cad96a7fee62ae7c229536f01ddd7f41241 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Mon, 13 Jul 2020 03:09:49 -0700 Subject: [PATCH] rename the function to match the rest of the test convention (#5692) --- tests/test_modeling_xlm.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_modeling_xlm.py b/tests/test_modeling_xlm.py index e805320da1..2a5cd4096a 100644 --- a/tests/test_modeling_xlm.py +++ b/tests/test_modeling_xlm.py @@ -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):