Add 'exclude' tuple to check_requirements() (#2041)
Browse files- utils/general.py +3 -3
utils/general.py
CHANGED
@@ -79,11 +79,11 @@ def check_git_status():
|
|
79 |
print(e)
|
80 |
|
81 |
|
82 |
-
def check_requirements(file='requirements.txt'):
|
83 |
# Check installed dependencies meet requirements
|
84 |
import pkg_resources
|
85 |
-
requirements = pkg_resources.parse_requirements(Path(file).open())
|
86 |
-
|
87 |
pkg_resources.require(requirements) # DistributionNotFound or VersionConflict exception if requirements not met
|
88 |
|
89 |
|
|
|
79 |
print(e)
|
80 |
|
81 |
|
82 |
+
def check_requirements(file='requirements.txt', exclude=()):
|
83 |
# Check installed dependencies meet requirements
|
84 |
import pkg_resources
|
85 |
+
requirements = [f'{x.name}{x.specifier}' for x in pkg_resources.parse_requirements(Path(file).open())
|
86 |
+
if x.name not in exclude]
|
87 |
pkg_resources.require(requirements) # DistributionNotFound or VersionConflict exception if requirements not met
|
88 |
|
89 |
|