From dd816f3675fa3998d74cedc8af3873f982389618 Mon Sep 17 00:00:00 2001 From: Christian Clauss Date: Tue, 12 Nov 2024 04:53:04 +0100 Subject: [PATCH] Replace pkg_resources.parse_requirements with packaging.requirements.Requiremen --- setup.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/setup.py b/setup.py index 0226320..4d91a5a 100644 --- a/setup.py +++ b/setup.py @@ -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"], },