forgot to add regex to requirements :(

This commit is contained in:
thomwolf
2019-02-18 11:54:51 +01:00
parent ab7f5d2943
commit e0855e8929
3 changed files with 7 additions and 4 deletions

View File

@@ -1,4 +1,4 @@
__version__ = "0.6.0" __version__ = "0.6.1"
from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer
from .tokenization_openai import OpenAIGPTTokenizer from .tokenization_openai import OpenAIGPTTokenizer
from .tokenization_transfo_xl import (TransfoXLTokenizer, TransfoXLCorpus) from .tokenization_transfo_xl import (TransfoXLTokenizer, TransfoXLCorpus)

View File

@@ -6,3 +6,5 @@ tqdm
boto3 boto3
# Used for downloading models over HTTP # Used for downloading models over HTTP
requests requests
# For OpenAI GPT
regex

View File

@@ -38,7 +38,7 @@ from setuptools import find_packages, setup
setup( setup(
name="pytorch_pretrained_bert", name="pytorch_pretrained_bert",
version="0.6.0", version="0.6.1",
author="Thomas Wolf, Victor Sanh, Tim Rault, Google AI Language Team Authors, Open AI team Authors", author="Thomas Wolf, Victor Sanh, Tim Rault, Google AI Language Team Authors, Open AI team Authors",
author_email="thomas@huggingface.co", author_email="thomas@huggingface.co",
description="PyTorch version of Google AI BERT model with script to load Google pre-trained models", description="PyTorch version of Google AI BERT model with script to load Google pre-trained models",
@@ -53,7 +53,8 @@ setup(
'numpy', 'numpy',
'boto3', 'boto3',
'requests', 'requests',
'tqdm'], 'tqdm',
'regex'],
entry_points={ entry_points={
'console_scripts': [ 'console_scripts': [
"pytorch_pretrained_bert=pytorch_pretrained_bert.__main__:main", "pytorch_pretrained_bert=pytorch_pretrained_bert.__main__:main",