Преглед изворни кода

update paddlehub requirements

jiangjiajun пре 4 година
родитељ
комит
2776149603
2 измењених фајлова са 7 додато и 3 уклоњено
  1. 5 1
      paddlex/__init__.py
  2. 2 2
      setup.py

+ 5 - 1
paddlex/__init__.py

@@ -14,7 +14,7 @@
 
 from __future__ import absolute_import
 
-__version__ = '1.3.4'
+__version__ = '1.3.5'
 
 import os
 if 'FLAGS_eager_delete_tensor_gb' not in os.environ:
@@ -32,6 +32,10 @@ if version[0] == '1':
         raise Exception(
             'For running paddlex(v{}), Version of paddlepaddle should be greater than 1.8.3'.
             format(__version__))
+    import paddlehub as hub
+    if hub.__version__.strip().split('.')[0] > '1':
+        raise Exception("Try to reinstall Paddlehub by 'pip install paddlehub==1.8.2' while paddlepaddle < 2.0")
+
 if hasattr(paddle, 'enable_static'):
     paddle.enable_static()
 

+ 2 - 2
setup.py

@@ -19,7 +19,7 @@ long_description = "PaddlePaddle Entire Process Development Toolkit"
 
 setuptools.setup(
     name="paddlex",
-    version='1.3.4',
+    version='1.3.5',
     author="paddlex",
     author_email="paddlex@baidu.com",
     description=long_description,
@@ -39,7 +39,7 @@ setuptools.setup(
     setup_requires=['cython', 'numpy'],
     install_requires=[
         "pycocotools;platform_system!='Windows'", 'pyyaml', 'colorama', 'tqdm',
-        'paddleslim==1.1.1', 'visualdl>=2.0.0', 'paddlehub>=1.8.2',
+        'paddleslim==1.1.1', 'visualdl>=2.0.0', 'paddlehub==1.8.2',
         'shapely>=1.7.0', 'opencv-python', 'flask_cors', 'sklearn', 'psutil',
         'xlwt'
     ],