Skip to content

GitLab

Commit 6dc4b6f3 authored by thomwolf's avatar thomwolf
Browse files

skip transfo-xl tokenizer tests with tf for now

parent e30579f7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
pytorch_transformers/tests/tokenization_transfo_xl_test.py
View file @ 6dc4b6f3
...@@ -22,6 +22,7 @@ from io import open ...@@ -22,6 +22,7 @@ from io import open
from pytorch_transformers import is_torch_available from pytorch_transformers import is_torch_available
try: try:
import torch
from pytorch_transformers.tokenization_transfo_xl import TransfoXLTokenizer, VOCAB_FILES_NAMES from pytorch_transformers.tokenization_transfo_xl import TransfoXLTokenizer, VOCAB_FILES_NAMES
except ImportError: except ImportError:
pytestmark = pytest.mark.skip("Require Torch") # TODO: untangle Transfo-XL tokenizer from torch.load and torch.save pytestmark = pytest.mark.skip("Require Torch") # TODO: untangle Transfo-XL tokenizer from torch.load and torch.save
......
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