2121import warnings
2222import platform
2323import subprocess # nosec B404 # disable import-subprocess check
24- from distutils .version import LooseVersion
24+ from packaging .version import Version
2525from pathlib import Path
2626from setuptools import find_packages , setup # pylint:disable=W9902
2727from setuptools .command .test import test as test_command # pylint:disable=W9902
@@ -55,7 +55,7 @@ def read(*path):
5555def check_and_update_numpy (min_acceptable = '1.15' ):
5656 try :
5757 import numpy as np # pylint:disable=C0415
58- update_required = LooseVersion (np .__version__ ) < LooseVersion (min_acceptable )
58+ update_required = Version (np .__version__ ) < Version (min_acceptable )
5959 except ImportError :
6060 update_required = True
6161 if update_required :
@@ -106,6 +106,12 @@ def prepare_requirements():
106106 check_and_update_numpy ()
107107
108108
109+
110+ _extras = _extras + ['pycocotools>=2.0.2' , 'torch>=0.4.0' , 'torchvision>=0.2.1' , 'lpips' , 'soundfile' , "torchmetrics" , "diffusers" ]
111+ if sys .version_info .major == 3 and sys .version_info .minor < 12 :
112+ _extras .append ('crf_beam;platform_system=="Linux"' )
113+ _extras .append ('kenlm @ git+https://github.com/kpu/kenlm.git@4cb443e60b7bf2c0ddf3c745378f76cb59e254e5#egg=kenlm' )
114+
109115setup (
110116 name = "accuracy_checker" ,
111117 description = "Deep Learning Accuracy validation framework" ,
@@ -121,6 +127,5 @@ def prepare_requirements():
121127 install_requires = _requirements ,
122128 tests_require = [read ("requirements-test.in" )],
123129 cmdclass = {'test' : PyTest , 'install_core' : CoreInstall },
124- extras_require = {'extra' : _extras + ['pycocotools>=2.0.2' , 'crf_beam;platform_system=="Linux"' , 'torch>=0.4.0' , 'torchvision>=0.2.1' , 'lpips' , 'soundfile' , "torchmetrics" , "diffusers" ,
125- 'kenlm @ git+https://github.com/kpu/kenlm.git@4cb443e60b7bf2c0ddf3c745378f76cb59e254e5#egg=kenlm' ]}
130+ extras_require = {'extra' : _extras }
126131)
0 commit comments