Add check_requirements() (#1853)

* Add check_requirements()

* add import

* parameterize filename

* add to detect, test
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-01-06 16:35:40 -08:00
1 parent 135ec5c5ce
commit e77c77f580
4 files changed
+16 -5

No files matched your search

+8
View File
@@ -53,6 +53,14 @@ def check_git_status():
print(s[s.find('Your branch is behind'):s.find('\n\n')] + '\n')
def check_requirements(file='requirements.txt'):
# 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]
pkg_resources.require(requirements) # DistributionNotFound or VersionConflict exception if requirements not met
def check_img_size(img_size, s=32):
# Verify img_size is a multiple of stride s
new_size = make_divisible(img_size, int(s)) # ceil gs-multiple