diff --git a/examples/summarization/test_utils_summarization.py b/examples/summarization/test_utils_summarization.py index 86ec5b6006..d562ad04b7 100644 --- a/examples/summarization/test_utils_summarization.py +++ b/examples/summarization/test_utils_summarization.py @@ -98,7 +98,3 @@ class SummarizationDataProcessingTest(unittest.TestCase): result = compute_token_type_ids(batch, separator) np.testing.assert_array_equal(result, expected) - - -if __name__ == "__main__": - unittest.main() diff --git a/examples/test_examples.py b/examples/test_examples.py index d27f5671a4..8a01b4b0bc 100644 --- a/examples/test_examples.py +++ b/examples/test_examples.py @@ -104,7 +104,3 @@ class ExamplesTests(unittest.TestCase): with patch.object(sys, "argv", testargs + [model_type, model_name]): result = run_generation.main() self.assertGreaterEqual(len(result), 10) - - -if __name__ == "__main__": - unittest.main() diff --git a/templates/adding_a_new_model/tests/test_modeling_tf_xxx.py b/templates/adding_a_new_model/tests/test_modeling_tf_xxx.py index 8879c7e7d1..4a1d66b591 100644 --- a/templates/adding_a_new_model/tests/test_modeling_tf_xxx.py +++ b/templates/adding_a_new_model/tests/test_modeling_tf_xxx.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import XxxConfig, is_tf_available from .test_configuration_common import ConfigTester @@ -251,7 +249,3 @@ class TFXxxModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in ["xxx-base-uncased"]: model = TFXxxModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/templates/adding_a_new_model/tests/test_modeling_xxx.py b/templates/adding_a_new_model/tests/test_modeling_xxx.py index e49b8e435b..b6364447b4 100644 --- a/templates/adding_a_new_model/tests/test_modeling_xxx.py +++ b/templates/adding_a_new_model/tests/test_modeling_xxx.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -272,7 +270,3 @@ class XxxModelTest(CommonTestCases.CommonModelTester): for model_name in list(XXX_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = XxxModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/templates/adding_a_new_model/tests/test_tokenization_xxx.py b/templates/adding_a_new_model/tests/test_tokenization_xxx.py index 087c1002d1..0f46cfa3a3 100644 --- a/templates/adding_a_new_model/tests/test_tokenization_xxx.py +++ b/templates/adding_a_new_model/tests/test_tokenization_xxx.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function, unicode_literals import os -import unittest from io import open from transformers.tokenization_bert import VOCAB_FILES_NAMES, XxxTokenizer @@ -63,7 +62,3 @@ class XxxTokenizationTest(CommonTestCases.CommonTokenizerTester): tokens = tokenizer.tokenize("UNwant\u00E9d,running") self.assertListEqual(tokens, ["un", "##want", "##ed", ",", "runn", "##ing"]) self.assertListEqual(tokenizer.convert_tokens_to_ids(tokens), [7, 4, 5, 10, 8, 9]) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_configuration_common.py b/tests/test_configuration_common.py index 234301df21..a94ddaea7b 100644 --- a/tests/test_configuration_common.py +++ b/tests/test_configuration_common.py @@ -16,7 +16,6 @@ from __future__ import absolute_import, division, print_function import json import os -import unittest from .tokenization_tests_commons import TemporaryDirectory @@ -64,7 +63,3 @@ class ConfigTester(object): self.create_and_test_config_to_json_string() self.create_and_test_config_to_json_file() self.create_and_test_config_from_and_save_pretrained() - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py index af72408d29..e6a51a77f0 100644 --- a/tests/test_hf_api.py +++ b/tests/test_hf_api.py @@ -102,7 +102,3 @@ class HfFolderTest(unittest.TestCase): # ^^ not an error, we test that the # second call does not fail. self.assertEqual(HfFolder.get_token(), None) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_model_card.py b/tests/test_model_card.py index 9b9947a720..9fcae1dbd8 100644 --- a/tests/test_model_card.py +++ b/tests/test_model_card.py @@ -80,7 +80,3 @@ class ModelCardTester(unittest.TestCase): model_card_second = ModelCard.from_pretrained(tmpdirname) self.assertEqual(model_card_second.to_dict(), model_card_first.to_dict()) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_albert.py b/tests/test_modeling_albert.py index d0d85e670a..eeecb0d4d4 100644 --- a/tests/test_modeling_albert.py +++ b/tests/test_modeling_albert.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -249,7 +247,3 @@ class AlbertModelTest(CommonTestCases.CommonModelTester): for model_name in list(ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = AlbertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_auto.py b/tests/test_modeling_auto.py index a174dca86a..ab6ae3c691 100644 --- a/tests/test_modeling_auto.py +++ b/tests/test_modeling_auto.py @@ -100,7 +100,3 @@ class AutoModelTest(unittest.TestCase): logging.basicConfig(level=logging.INFO) model = AutoModelWithLMHead.from_pretrained(SMALL_MODEL_IDENTIFIER) self.assertIsInstance(model, BertForMaskedLM) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_bert.py b/tests/test_modeling_bert.py index f96993bb50..52aaece535 100644 --- a/tests/test_modeling_bert.py +++ b/tests/test_modeling_bert.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -475,7 +473,3 @@ class BertModelTest(CommonTestCases.CommonModelTester): for model_name in list(BERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = BertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 591aa648c5..0d1cfbd311 100644 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -892,7 +892,3 @@ class ModelUtilsTest(unittest.TestCase): self.assertEqual(model.config.output_attentions, True) self.assertEqual(model.config.output_hidden_states, True) self.assertEqual(model.config, config) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_ctrl.py b/tests/test_modeling_ctrl.py index 7d2d2e12b6..519b529299 100644 --- a/tests/test_modeling_ctrl.py +++ b/tests/test_modeling_ctrl.py @@ -13,8 +13,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -211,7 +209,3 @@ class CTRLModelTest(CommonTestCases.CommonModelTester): for model_name in list(CTRL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = CTRLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_distilbert.py b/tests/test_modeling_distilbert.py index e432761c1b..6c83751bad 100644 --- a/tests/test_modeling_distilbert.py +++ b/tests/test_modeling_distilbert.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -250,7 +248,3 @@ class DistilBertModelTest(CommonTestCases.CommonModelTester): # for model_name in list(DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: # model = DistilBertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) # self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_encoder_decoder.py b/tests/test_modeling_encoder_decoder.py index b9cef6667a..ac01e7b561 100644 --- a/tests/test_modeling_encoder_decoder.py +++ b/tests/test_modeling_encoder_decoder.py @@ -48,7 +48,3 @@ class EncoderDecoderModelTest(unittest.TestCase): with self.assertRaises(ValueError): _ = Model2Model.from_pretrained("does-not-exist") - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_gpt2.py b/tests/test_modeling_gpt2.py index daa6ed9269..2ca8f14d7a 100644 --- a/tests/test_modeling_gpt2.py +++ b/tests/test_modeling_gpt2.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -248,7 +246,3 @@ class GPT2ModelTest(CommonTestCases.CommonModelTester): for model_name in list(GPT2_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = GPT2Model.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_openai.py b/tests/test_modeling_openai.py index 1fd198577a..d9b2d0f228 100644 --- a/tests/test_modeling_openai.py +++ b/tests/test_modeling_openai.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -205,7 +203,3 @@ class OpenAIGPTModelTest(CommonTestCases.CommonModelTester): for model_name in list(OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = OpenAIGPTModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_roberta.py b/tests/test_modeling_roberta.py index 2141082e4d..61d2055aea 100644 --- a/tests/test_modeling_roberta.py +++ b/tests/test_modeling_roberta.py @@ -298,7 +298,3 @@ class RobertaModelIntegrationTest(unittest.TestCase): self.assertEqual(output.shape, expected_shape) expected_tensor = torch.Tensor([[-0.9469, 0.3913, 0.5118]]) self.assertTrue(torch.allclose(output, expected_tensor, atol=1e-3)) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_t5.py b/tests/test_modeling_t5.py index 76286dbd36..f12fa10d3b 100644 --- a/tests/test_modeling_t5.py +++ b/tests/test_modeling_t5.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -212,7 +210,3 @@ class T5ModelTest(CommonTestCases.CommonModelTester): for model_name in list(T5_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = T5Model.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_albert.py b/tests/test_modeling_tf_albert.py index 0a569a467c..3761c3252f 100644 --- a/tests/test_modeling_tf_albert.py +++ b/tests/test_modeling_tf_albert.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import AlbertConfig, is_tf_available from .test_configuration_common import ConfigTester @@ -213,7 +211,3 @@ class TFAlbertModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in list(TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TFAlbertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_auto.py b/tests/test_modeling_tf_auto.py index b06d52ed2d..cc9abaa03d 100644 --- a/tests/test_modeling_tf_auto.py +++ b/tests/test_modeling_tf_auto.py @@ -99,7 +99,3 @@ class TFAutoModelTest(unittest.TestCase): logging.basicConfig(level=logging.INFO) model = TFAutoModelWithLMHead.from_pretrained(SMALL_MODEL_IDENTIFIER) self.assertIsInstance(model, TFBertForMaskedLM) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_bert.py b/tests/test_modeling_tf_bert.py index cce488cf0e..d93fd133ee 100644 --- a/tests/test_modeling_tf_bert.py +++ b/tests/test_modeling_tf_bert.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import BertConfig, is_tf_available from .test_configuration_common import ConfigTester @@ -315,7 +313,3 @@ class TFBertModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in ["bert-base-uncased"]: model = TFBertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py index fb85181a7a..868eb2d9e0 100644 --- a/tests/test_modeling_tf_common.py +++ b/tests/test_modeling_tf_common.py @@ -378,7 +378,3 @@ def ids_tensor(shape, vocab_size, rng=None, name=None, dtype=None): output = tf.constant(values, shape=shape, dtype=dtype if dtype is not None else tf.int32) return output - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_ctrl.py b/tests/test_modeling_tf_ctrl.py index 763a6adb58..02845e60b5 100644 --- a/tests/test_modeling_tf_ctrl.py +++ b/tests/test_modeling_tf_ctrl.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import CTRLConfig, is_tf_available from .test_configuration_common import ConfigTester @@ -201,7 +199,3 @@ class TFCTRLModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in list(TF_CTRL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TFCTRLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_distilbert.py b/tests/test_modeling_tf_distilbert.py index e82436c026..3fa1ea849b 100644 --- a/tests/test_modeling_tf_distilbert.py +++ b/tests/test_modeling_tf_distilbert.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import DistilBertConfig, is_tf_available from .test_configuration_common import ConfigTester @@ -221,7 +219,3 @@ class TFDistilBertModelTest(TFCommonTestCases.TFCommonModelTester): # for model_name in list(DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: # model = DistilBertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) # self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_gpt2.py b/tests/test_modeling_tf_gpt2.py index c7e2334572..e97462258e 100644 --- a/tests/test_modeling_tf_gpt2.py +++ b/tests/test_modeling_tf_gpt2.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import GPT2Config, is_tf_available from .test_configuration_common import ConfigTester @@ -234,7 +232,3 @@ class TFGPT2ModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in list(TF_GPT2_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TFGPT2Model.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_openai_gpt.py b/tests/test_modeling_tf_openai_gpt.py index 5d4c7905a9..ea463c0e4f 100644 --- a/tests/test_modeling_tf_openai_gpt.py +++ b/tests/test_modeling_tf_openai_gpt.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import OpenAIGPTConfig, is_tf_available from .test_configuration_common import ConfigTester @@ -235,7 +233,3 @@ class TFOpenAIGPTModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in list(TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TFOpenAIGPTModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_roberta.py b/tests/test_modeling_tf_roberta.py index 6369b1d6f5..6318eeaabf 100644 --- a/tests/test_modeling_tf_roberta.py +++ b/tests/test_modeling_tf_roberta.py @@ -244,7 +244,3 @@ class TFRobertaModelIntegrationTest(unittest.TestCase): self.assertEqual(list(output.numpy().shape), expected_shape) expected_tensor = tf.constant([[-0.9469, 0.3913, 0.5118]]) self.assertTrue(numpy.allclose(output.numpy(), expected_tensor.numpy(), atol=1e-3)) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_t5.py b/tests/test_modeling_tf_t5.py index 0d2ccfe140..e18213c0b5 100644 --- a/tests/test_modeling_tf_t5.py +++ b/tests/test_modeling_tf_t5.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import T5Config, is_tf_available from .test_configuration_common import ConfigTester @@ -165,7 +163,3 @@ class TFT5ModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in ["t5-small"]: model = TFT5Model.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_transfo_xl.py b/tests/test_modeling_tf_transfo_xl.py index ed5753c11d..bc2d616fe2 100644 --- a/tests/test_modeling_tf_transfo_xl.py +++ b/tests/test_modeling_tf_transfo_xl.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function import random -import unittest from transformers import TransfoXLConfig, is_tf_available @@ -207,7 +206,3 @@ class TFTransfoXLModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in list(TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TFTransfoXLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_xlm.py b/tests/test_modeling_tf_xlm.py index 40a06ccfa4..0ee924ab61 100644 --- a/tests/test_modeling_tf_xlm.py +++ b/tests/test_modeling_tf_xlm.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_tf_available from .test_configuration_common import ConfigTester @@ -306,7 +304,3 @@ class TFXLMModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in list(TF_XLM_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TFXLMModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_tf_xlnet.py b/tests/test_modeling_tf_xlnet.py index c3456787dd..0dcd8a776a 100644 --- a/tests/test_modeling_tf_xlnet.py +++ b/tests/test_modeling_tf_xlnet.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function import random -import unittest from transformers import XLNetConfig, is_tf_available @@ -401,7 +400,3 @@ class TFXLNetModelTest(TFCommonTestCases.TFCommonModelTester): for model_name in list(TF_XLNET_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TFXLNetModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_transfo_xl.py b/tests/test_modeling_transfo_xl.py index 8465f351db..76886a83ad 100644 --- a/tests/test_modeling_transfo_xl.py +++ b/tests/test_modeling_transfo_xl.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function import random -import unittest from transformers import is_torch_available @@ -208,7 +207,3 @@ class TransfoXLModelTest(CommonTestCases.CommonModelTester): for model_name in list(TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = TransfoXLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_xlm.py b/tests/test_modeling_xlm.py index 8ac3b5050a..c921a4312f 100644 --- a/tests/test_modeling_xlm.py +++ b/tests/test_modeling_xlm.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function -import unittest - from transformers import is_torch_available from .test_configuration_common import ConfigTester @@ -390,7 +388,3 @@ class XLMModelTest(CommonTestCases.CommonModelTester): for model_name in list(XLM_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = XLMModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_modeling_xlnet.py b/tests/test_modeling_xlnet.py index 0ba7dd23b5..c295b79b28 100644 --- a/tests/test_modeling_xlnet.py +++ b/tests/test_modeling_xlnet.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function import random -import unittest from transformers import is_torch_available @@ -499,7 +498,3 @@ class XLNetModelTest(CommonTestCases.CommonModelTester): for model_name in list(XLNET_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: model = XLNetModel.from_pretrained(model_name, cache_dir=CACHE_DIR) self.assertIsNotNone(model) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_optimization.py b/tests/test_optimization.py index c0cef1e387..267fbf0986 100644 --- a/tests/test_optimization.py +++ b/tests/test_optimization.py @@ -150,7 +150,3 @@ class ScheduleInitTest(unittest.TestCase): ) lrs_2 = unwrap_and_save_reload_schedule(scheduler, self.num_steps) self.assertListEqual([l[0] for l in lrs], [l[0] for l in lrs_2]) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_optimization_tf.py b/tests/test_optimization_tf.py index 4058aaf835..bff667fd3f 100644 --- a/tests/test_optimization_tf.py +++ b/tests/test_optimization_tf.py @@ -83,7 +83,3 @@ class OptimizationFTest(unittest.TestCase): self.assertEqual(accumulator.step, 0) self.assertListAlmostEqual(accumulator._gradients[0].values[0].value().numpy().tolist(), [0.0, 0.0], tol=1e-2) self.assertListAlmostEqual(accumulator._gradients[0].values[1].value().numpy().tolist(), [0.0, 0.0], tol=1e-2) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py index 6f3ea19c7b..43cc1f2474 100644 --- a/tests/test_pipelines.py +++ b/tests/test_pipelines.py @@ -205,7 +205,3 @@ class MultiColumnInputTestCase(unittest.TestCase): for tokenizer, model, config in TF_QA_FINETUNED_MODELS: nlp = pipeline(task="question-answering", model=model, config=config, tokenizer=tokenizer) self._test_multicolumn_pipeline(nlp, valid_samples, invalid_samples, mandatory_output_keys) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_albert.py b/tests/test_tokenization_albert.py index 88d18031fd..d22e8760a9 100644 --- a/tests/test_tokenization_albert.py +++ b/tests/test_tokenization_albert.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function, unicode_literals import os -import unittest from transformers.tokenization_albert import AlbertTokenizer @@ -78,7 +77,3 @@ class AlbertTokenizationTest(CommonTestCases.CommonTokenizerTester): assert encoded_pair == [tokenizer.cls_token_id] + text + [tokenizer.sep_token_id] + text_2 + [ tokenizer.sep_token_id ] - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_auto.py b/tests/test_tokenization_auto.py index 929f5f8a6a..7afe543741 100644 --- a/tests/test_tokenization_auto.py +++ b/tests/test_tokenization_auto.py @@ -49,7 +49,3 @@ class AutoTokenizerTest(unittest.TestCase): tokenizer = AutoTokenizer.from_pretrained(SMALL_MODEL_IDENTIFIER) self.assertIsInstance(tokenizer, BertTokenizer) self.assertEqual(len(tokenizer), 12) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_bert.py b/tests/test_tokenization_bert.py index 9c8c18fe44..2081570fb3 100644 --- a/tests/test_tokenization_bert.py +++ b/tests/test_tokenization_bert.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function, unicode_literals import os -import unittest from io import open from transformers.tokenization_bert import ( @@ -146,7 +145,3 @@ class BertTokenizationTest(CommonTestCases.CommonTokenizerTester): assert encoded_sentence == [101] + text + [102] assert encoded_pair == [101] + text + [102] + text_2 + [102] - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_ctrl.py b/tests/test_tokenization_ctrl.py index eb3fbb9da4..612e01213f 100644 --- a/tests/test_tokenization_ctrl.py +++ b/tests/test_tokenization_ctrl.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera import json import os -import unittest from io import open from transformers.tokenization_ctrl import VOCAB_FILES_NAMES, CTRLTokenizer @@ -63,7 +62,3 @@ class CTRLTokenizationTest(CommonTestCases.CommonTokenizerTester): input_bpe_tokens = [0, 1, 2, 4, 5, 1, 0, 3, 6] self.assertListEqual(tokenizer.convert_tokens_to_ids(input_tokens), input_bpe_tokens) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_distilbert.py b/tests/test_tokenization_distilbert.py index cbaf546ddd..1e2084ee26 100644 --- a/tests/test_tokenization_distilbert.py +++ b/tests/test_tokenization_distilbert.py @@ -14,8 +14,6 @@ # limitations under the License. from __future__ import absolute_import, division, print_function, unicode_literals -import unittest - from transformers.tokenization_distilbert import DistilBertTokenizer from .test_tokenization_bert import BertTokenizationTest @@ -43,7 +41,3 @@ class DistilBertTokenizationTest(BertTokenizationTest): assert encoded_pair == [tokenizer.cls_token_id] + text + [tokenizer.sep_token_id] + text_2 + [ tokenizer.sep_token_id ] - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_gpt2.py b/tests/test_tokenization_gpt2.py index 9246e5ce17..0f916db3c9 100644 --- a/tests/test_tokenization_gpt2.py +++ b/tests/test_tokenization_gpt2.py @@ -16,7 +16,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera import json import os -import unittest from io import open from transformers.tokenization_gpt2 import VOCAB_FILES_NAMES, GPT2Tokenizer @@ -84,7 +83,3 @@ class GPT2TokenizationTest(CommonTestCases.CommonTokenizerTester): input_tokens = tokens + [tokenizer.unk_token] input_bpe_tokens = [14, 15, 10, 9, 3, 2, 15, 19] self.assertListEqual(tokenizer.convert_tokens_to_ids(input_tokens), input_bpe_tokens) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_openai.py b/tests/test_tokenization_openai.py index fe4ed77c13..c1a0fe10f7 100644 --- a/tests/test_tokenization_openai.py +++ b/tests/test_tokenization_openai.py @@ -16,7 +16,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera import json import os -import unittest from transformers.tokenization_openai import VOCAB_FILES_NAMES, OpenAIGPTTokenizer @@ -83,7 +82,3 @@ class OpenAIGPTTokenizationTest(CommonTestCases.CommonTokenizerTester): input_tokens = tokens + [""] input_bpe_tokens = [14, 15, 20] self.assertListEqual(tokenizer.convert_tokens_to_ids(input_tokens), input_bpe_tokens) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_roberta.py b/tests/test_tokenization_roberta.py index 92a1a6d5d5..9f70b1d27e 100644 --- a/tests/test_tokenization_roberta.py +++ b/tests/test_tokenization_roberta.py @@ -16,7 +16,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera import json import os -import unittest from io import open from transformers.tokenization_roberta import VOCAB_FILES_NAMES, RobertaTokenizer @@ -111,7 +110,3 @@ class RobertaTokenizationTest(CommonTestCases.CommonTokenizerTester): assert encoded_sentence == encoded_text_from_decode assert encoded_pair == encoded_pair_from_decode - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_t5.py b/tests/test_tokenization_t5.py index 69f209f290..e02d45af5f 100644 --- a/tests/test_tokenization_t5.py +++ b/tests/test_tokenization_t5.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function, unicode_literals import os -import unittest from transformers.tokenization_t5 import T5Tokenizer from transformers.tokenization_xlnet import SPIECE_UNDERLINE @@ -110,7 +109,3 @@ class T5TokenizationTest(CommonTestCases.CommonTokenizerTester): ".", ], ) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_transfo_xl.py b/tests/test_tokenization_transfo_xl.py index 6b1d1dfcec..df9d6cd6ab 100644 --- a/tests/test_tokenization_transfo_xl.py +++ b/tests/test_tokenization_transfo_xl.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function, unicode_literals import os -import unittest from io import open from transformers import is_torch_available @@ -83,7 +82,3 @@ class TransfoXLTokenizationTest(CommonTestCases.CommonTokenizerTester): self.assertListEqual( tokenizer.tokenize(" \tHeLLo ! how \n Are yoU ? "), ["HeLLo", "!", "how", "Are", "yoU", "?"] ) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_utils.py b/tests/test_tokenization_utils.py index 8865110663..d99ba2733f 100644 --- a/tests/test_tokenization_utils.py +++ b/tests/test_tokenization_utils.py @@ -44,7 +44,3 @@ class TokenizerUtilsTest(unittest.TestCase): @slow def test_pretrained_tokenizers(self): self.check_tokenizer_from_pretrained(GPT2Tokenizer) - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_xlm.py b/tests/test_tokenization_xlm.py index 3ce5015353..fe8cf1c974 100644 --- a/tests/test_tokenization_xlm.py +++ b/tests/test_tokenization_xlm.py @@ -16,7 +16,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera import json import os -import unittest from transformers.tokenization_xlm import VOCAB_FILES_NAMES, XLMTokenizer @@ -98,7 +97,3 @@ class XLMTokenizationTest(CommonTestCases.CommonTokenizerTester): assert encoded_sentence == [1] + text + [1] assert encoded_pair == [1] + text + [1] + text_2 + [1] - - -if __name__ == "__main__": - unittest.main() diff --git a/tests/test_tokenization_xlnet.py b/tests/test_tokenization_xlnet.py index 2c55a337ba..1dca9cef0e 100644 --- a/tests/test_tokenization_xlnet.py +++ b/tests/test_tokenization_xlnet.py @@ -15,7 +15,6 @@ from __future__ import absolute_import, division, print_function, unicode_literals import os -import unittest from transformers.tokenization_xlnet import SPIECE_UNDERLINE, XLNetTokenizer @@ -183,7 +182,3 @@ class XLNetTokenizationTest(CommonTestCases.CommonTokenizerTester): assert encoded_sentence == text + [4, 3] assert encoded_pair == text + [4] + text_2 + [4, 3] - - -if __name__ == "__main__": - unittest.main()