From 04d2006f28a6746e0b0521aa27b56328a6a66293 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Thu, 5 Sep 2019 11:22:13 +0200 Subject: [PATCH] skip transfo-xl tokenizer tests with tf for now --- pytorch_transformers/tests/tokenization_transfo_xl_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_transformers/tests/tokenization_transfo_xl_test.py b/pytorch_transformers/tests/tokenization_transfo_xl_test.py index 1406bac48b4..792033d82c3 100644 --- a/pytorch_transformers/tests/tokenization_transfo_xl_test.py +++ b/pytorch_transformers/tests/tokenization_transfo_xl_test.py @@ -22,6 +22,7 @@ from io import open from pytorch_transformers import is_torch_available try: + import torch from pytorch_transformers.tokenization_transfo_xl import TransfoXLTokenizer, VOCAB_FILES_NAMES except ImportError: pytestmark = pytest.mark.skip("Require Torch") # TODO: untangle Transfo-XL tokenizer from torch.load and torch.save