From e0855e89291dc0e2e65d4389992a21cda8dcc68f Mon Sep 17 00:00:00 2001
From: thomwolf <thomwolf@gmail.com>
Date: Mon, 18 Feb 2019 11:54:51 +0100
Subject: [PATCH] forgot to add regex to requirements :(

---
 pytorch_pretrained_bert/__init__.py | 2 +-
 requirements.txt                    | 4 +++-
 setup.py                            | 5 +++--
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/pytorch_pretrained_bert/__init__.py b/pytorch_pretrained_bert/__init__.py
index 7cb28e2d2b..bd455b8d9c 100644
--- a/pytorch_pretrained_bert/__init__.py
+++ b/pytorch_pretrained_bert/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.6.0"
+__version__ = "0.6.1"
 from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer
 from .tokenization_openai import OpenAIGPTTokenizer
 from .tokenization_transfo_xl import (TransfoXLTokenizer, TransfoXLCorpus)
diff --git a/requirements.txt b/requirements.txt
index f37f11cc54..caf6471e86 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,4 +5,6 @@ tqdm
 # Accessing files from S3 directly.
 boto3
 # Used for downloading models over HTTP
-requests
\ No newline at end of file
+requests
+# For OpenAI GPT
+regex
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 4070e35aa8..4cf9739a4b 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,7 @@ from setuptools import find_packages, setup
 
 setup(
     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_email="thomas@huggingface.co",
     description="PyTorch version of Google AI BERT model with script to load Google pre-trained models",
@@ -53,7 +53,8 @@ setup(
                       'numpy',
                       'boto3',
                       'requests',
-                      'tqdm'],
+                      'tqdm',
+                      'regex'],
     entry_points={
       'console_scripts': [
         "pytorch_pretrained_bert=pytorch_pretrained_bert.__main__:main",