Fix `check_requirements()` console output (#3738)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
single_channel
Glenn Jocher 1 year ago committed by GitHub
parent fadbef84e1
commit 71f78a62b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -223,7 +223,7 @@ def check_requirements(requirements=ROOT.parent / 'requirements.txt', exclude=()
requirements = [requirements] requirements = [requirements]
s = '' # console string s = '' # console string
n = 0 # number of packages updates pkgs = []
for r in requirements: for r in requirements:
rmin = r.split('/')[-1].replace('.git', '') # replace git+https://org/repo.git -> 'repo' rmin = r.split('/')[-1].replace('.git', '') # replace git+https://org/repo.git -> 'repo'
try: try:
@ -234,11 +234,11 @@ def check_requirements(requirements=ROOT.parent / 'requirements.txt', exclude=()
importlib.import_module(next(pkg.parse_requirements(rmin)).name) importlib.import_module(next(pkg.parse_requirements(rmin)).name)
except ImportError: except ImportError:
s += f'"{r}" ' s += f'"{r}" '
n += 1 pkgs.append(r)
if s: if s:
if install and AUTOINSTALL: # check environment variable if install and AUTOINSTALL: # check environment variable
pkgs = file or requirements # missing packages n = len(pkgs) # number of packages updates
LOGGER.info(f"{prefix} Ultralytics requirement{'s' * (n > 1)} {pkgs} not found, attempting AutoUpdate...") LOGGER.info(f"{prefix} Ultralytics requirement{'s' * (n > 1)} {pkgs} not found, attempting AutoUpdate...")
try: try:
t = time.time() t = time.time()

Loading…
Cancel
Save