Backwards compatability for args starting with data/

neuralmagic · Mar 2, 2023 · 88aea22 · 88aea22
1 parent ba97120
commit 88aea22
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/utils/general.py b/utils/general.py
@@ -455,6 +455,9 @@ def check_file(file, suffix=''):
             yaml_name = os.path.join(*file.split(os.sep)[1:])
             files.extend(glob.glob(str(ROOT / "models_v5.0" / '**' / yaml_name), recursive=True))
 
+        if file.startswith("data"):
+            file = os.path.join(*file.split(os.sep)[1:])
+
         for d in 'data', 'models', 'utils':  # search directories
             files.extend(glob.glob(str(ROOT / d / '**' / file), recursive=True))  # find file
         assert len(files), f'File not found: {file}'  # assert file was found