Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 14:35:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replace pkg_resources.parse_requirements with packaging.requirements.Requiremen

Browse Source
This commit is contained in:
Christian Clauss 2024-11-12 04:53:04 +01:00 committed by GitHub
parent f7e34da3e9
commit dd816f3675
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
setup.py
View File

@ -3,6 +3,7 @@ import platform
import sys
from pathlib import Path
from packaging.requirements import Requirement
from setuptools import find_packages, setup
@ -13,6 +14,11 @@ def read_version(fname="whisper/version.py"):
return version_module.__version__
def parse_requirements(file_path = Path(__file__).with_name("requirements.txt")):
lines = file_path.read_text(encoding="utf-8").splitlines()
return [str(Requirement(line)) for line in lines if not line.startswith("#")]
requirements = []
if sys.platform.startswith("linux") and platform.machine() == "x86_64":
requirements.append("triton>=2.0.0")
@ -30,12 +36,7 @@ setup(
url="https://github.com/openai/whisper",
license="MIT",
packages=find_packages(exclude=["tests*"]),
install_requires=[
str(r)
for r in pkg_resources.parse_requirements(
Path(__file__).with_name("requirements.txt").open()
)
],
install_requires=parse_requirements(Path(__file__).with_name("requirements.txt")),
entry_points={
"console_scripts": ["whisper=whisper.transcribe:cli"],
},