glenn-jocher commited on
Commit
f59f801
·
unverified ·
1 Parent(s): 59c21c7

Add 'exclude' tuple to check_requirements() (#2041)

Browse files
Files changed (1) hide show
  1. 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
- requirements = [x.name + ''.join(*x.specs) if len(x.specs) else x.name for x in requirements]
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