From 921f66f5b8b00e9e1267676b353b35cf31ca2e8a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 26 Jan 2021 21:17:36 -0800 Subject: [PATCH] Add 'exclude' tuple to check_requirements() (#2041) --- utils/general.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/general.py b/utils/general.py index 27cefe58a045..bbc0f32b8425 100755 --- a/utils/general.py +++ b/utils/general.py @@ -79,11 +79,11 @@ def check_git_status(): print(e) -def check_requirements(file='requirements.txt'): +def check_requirements(file='requirements.txt', exclude=()): # Check installed dependencies meet requirements import pkg_resources - requirements = pkg_resources.parse_requirements(Path(file).open()) - requirements = [x.name + ''.join(*x.specs) if len(x.specs) else x.name for x in requirements] + requirements = [f'{x.name}{x.specifier}' for x in pkg_resources.parse_requirements(Path(file).open()) + if x.name not in exclude] pkg_resources.require(requirements) # DistributionNotFound or VersionConflict exception if requirements not met