Update TensorFlow dependency of CkipTagger.

ckiplab · May 11, 2020 · ea20039 · ea20039
1 parent dd00dbe
commit ea20039
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/README.rst b/README.rst
@@ -92,9 +92,9 @@ Requirements
 
 * `Python <https://www.python.org>`_ 3.6+
 * `TreeLib <https://treelib.readthedocs.io>`_ 1.5+
-* `Tensorflow <https://www.tensorflow.org/>`_ 1.15
 * `CkipTagger <https://pypi.org/project/ckiptagger>`_ 0.1.1+ [Optional, Recommended]
 * `CkipClassic <https://ckip-classic.readthedocs.io>`_ 1.0+ [Optional]
+* `TensorFlow / TensorFlow-GPU <https://www.tensorflow.org/>`_ 1.13.1+, <2 [Required by CkipTagger]
 
 Driver Requirements
 ^^^^^^^^^^^^^^^^^^^
@@ -117,7 +117,7 @@ Installation via Pip
 ^^^^^^^^^^^^^^^^^^^^
 
 - No backend (not recommended): ``pip install ckipnlp``.
-- With CkipTagger backend (recommended): ``pip install ckipnlp[tagger]``
+- With CkipTagger backend (recommended): ``pip install ckipnlp[tagger]`` or ``pip install ckipnlp[tagger-gpu]``.
 - With CkipClassic backend: Please refer https://ckip-classic.readthedocs.io/en/latest/main/readme.html#installation for CkipClassic installation guide.
 
 Usage

diff --git a/ckipnlp/__init__.py b/ckipnlp/__init__.py
@@ -10,7 +10,7 @@
 __copyright__ = '2018-2020 CKIP Lab'
 
 __title__ = 'CKIPNLP'
-__version__ = '0.8.3'
+__version__ = '0.8.4dev'
 __description__ = 'CKIP CoreNLP'
 __license__ = 'CC BY-NC-SA 4.0'
 

diff --git a/setup.py b/setup.py
@@ -47,11 +47,11 @@ def main():
             'appdirs>=1.4.3',
             'numpy>=1.18',
             'treelib>=1.5.5',
-            'tensorflow==1.15'
         ],
         extras_require={
             # 'classic': ['ckip-classic>=1.0'],
-            'tagger': ['ckiptagger>=0.1.1'],
+            'tagger': ['ckiptagger[tf]>=0.1.1'],
+            'tagger-gpu': ['ckiptagger[tfgpu]>=0.1.1'],
         },
         data_files=[],
     )