瀏覽代碼

update requirements

will-jl944 4 年之前
父節點
當前提交
5c9769b45a
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      dygraph/setup.py

+ 1 - 1
dygraph/setup.py

@@ -31,7 +31,7 @@ setuptools.setup(
     install_requires=[
         "pycocotools;platform_system!='Windows'", 'pyyaml', 'colorama', 'tqdm',
         'paddleslim==2.1.0', 'visualdl>=2.1.1', 'shapely>=1.7.0',
-        'opencv-python', 'scipy', 'lap', 'motmetrics'
+        'opencv-python', 'scipy', 'lap', 'motmetrics', 'scikit-learn==0.23.2'
     ],
     classifiers=[
         "Programming Language :: Python :: 3",