Improved check_suffix() robustness to '' and "" (#5192)
* Improved check_suffix() robustness to `''` and `""` * Cleanup
This commit is contained in:
1 parent
0be58f16f0
commit
7ee5aed0b3
1 file changed
+4
-2
+4
-2
@@ -293,12 +293,14 @@ def check_imshow():
|
|||||||
|
|
||||||
|
|
||||||
def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''):
|
def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''):
|
||||||
# Check file(s) for acceptable suffixes
|
# Check file(s) for acceptable suffix
|
||||||
if file and suffix:
|
if file and suffix:
|
||||||
if isinstance(suffix, str):
|
if isinstance(suffix, str):
|
||||||
suffix = [suffix]
|
suffix = [suffix]
|
||||||
for f in file if isinstance(file, (list, tuple)) else [file]:
|
for f in file if isinstance(file, (list, tuple)) else [file]:
|
||||||
assert Path(f).suffix.lower() in suffix, f"{msg}{f} acceptable suffix is {suffix}"
|
s = Path(f).suffix.lower() # file suffix
|
||||||
|
if len(s):
|
||||||
|
assert s in suffix, f"{msg}{f} acceptable suffix is {suffix}"
|
||||||
|
|
||||||
|
|
||||||
def check_yaml(file, suffix=('.yaml', '.yml')):
|
def check_yaml(file, suffix=('.yaml', '.yml')):
|
||||||
|
|||||||
Reference in new issue
Block a user