From 39c17ce0b922bbafcf9f8da64f286fef01040727 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 25 Sep 2021 09:16:14 -0700 Subject: [PATCH] Fix `root` referenced before assignment (#4920) * Fix `root` referenced before assignment Fix for bug introduced by #4919 discovered on VOC autodownload: ``` python train.py --data VOC.yaml ``` * Cleanup --- utils/general.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/general.py b/utils/general.py index 2e2cdf389075..3c5cbff13d55 100755 --- a/utils/general.py +++ b/utils/general.py @@ -352,11 +352,11 @@ def check_dataset(data, autodownload=True): if not all(x.exists() for x in val): print('\nWARNING: Dataset not found, nonexistent paths: %s' % [str(x) for x in val if not x.exists()]) if s and autodownload: # download script + root = path.parent if 'path' in data else '..' # unzip directory i.e. '../' if s.startswith('http') and s.endswith('.zip'): # URL f = Path(s).name # filename print(f'Downloading {s} to {f}...') torch.hub.download_url_to_file(s, f) - root = path.parent if 'path' in data else '..' # unzip directory i.e. '../' Path(root).mkdir(parents=True, exist_ok=True) # create root ZipFile(f).extractall(path=root) # unzip Path(f).unlink() # remove zip @@ -366,7 +366,7 @@ def check_dataset(data, autodownload=True): r = os.system(s) else: # python script r = exec(s, {'yaml': data}) # return None - print(f"Dataset autodownload {f'success, saved to {root}' if r in (0, None) else 'failure'}") + print(f"Dataset autodownload {f'success, saved to {root}' if r in (0, None) else 'failure'}\n") else: raise Exception('Dataset not found.')