From 4480c51b1f220eb4fd94f8f862bcc2b0e6cd9aa4 Mon Sep 17 00:00:00 2001 From: kinghuin Date: Thu, 21 Nov 2019 13:03:41 +0800 Subject: [PATCH] fix requirement and setup --- requirements.txt | 1 + requirements_py2.txt | 1 + setup.py | 3 ++- 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 5866e150..9264b782 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,3 +14,4 @@ flake8 tb-paddle tensorboard >= 1.15 cma == 2.7.0 +sentencepiece diff --git a/requirements_py2.txt b/requirements_py2.txt index e7e54a3f..c3981c3a 100644 --- a/requirements_py2.txt +++ b/requirements_py2.txt @@ -13,3 +13,4 @@ flask >= 1.1.0 tb-paddle tensorboard >= 1.15 cma == 2.7.0 +sentencepiece diff --git a/setup.py b/setup.py index 8a7d5246..4a5cc342 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,8 @@ max_version, mid_version, min_version = python_version() REQUIRED_PACKAGES = [ 'six >= 1.10.0', 'protobuf >= 3.6.0', 'pyyaml', 'Pillow', 'requests', - 'tb-paddle', 'tensorboard >= 1.15', 'cma == 2.7.0', 'flask >= 1.1.0' + 'tb-paddle', 'tensorboard >= 1.15', 'cma == 2.7.0', 'flask >= 1.1.0', + 'sentencepiece' ] if max_version < 3: -- GitLab