Skip to content

Commit 88aea22

Browse files
committed
Backwards compatability for args starting with data/
1 parent ba97120 commit 88aea22

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

utils/general.py

+3
Original file line numberDiff line numberDiff line change
@@ -455,6 +455,9 @@ def check_file(file, suffix=''):
455455
yaml_name = os.path.join(*file.split(os.sep)[1:])
456456
files.extend(glob.glob(str(ROOT / "models_v5.0" / '**' / yaml_name), recursive=True))
457457

458+
if file.startswith("data"):
459+
file = os.path.join(*file.split(os.sep)[1:])
460+
458461
for d in 'data', 'models', 'utils': # search directories
459462
files.extend(glob.glob(str(ROOT / d / '**' / file), recursive=True)) # find file
460463
assert len(files), f'File not found: {file}' # assert file was found

0 commit comments

Comments
 (0)