diff --git a/nanodet/data/dataset/yolo.py b/nanodet/data/dataset/yolo.py index 61dd7147f..f6b75b986 100644 --- a/nanodet/data/dataset/yolo.py +++ b/nanodet/data/dataset/yolo.py @@ -53,8 +53,8 @@ def __init__(self, class_names, **kwargs): @staticmethod def _find_image( - image_prefix: str, - image_types: Sequence[str] = (".png", ".jpg", ".jpeg", ".bmp", ".tiff"), + image_prefix: str, + image_types: Sequence[str] = (".png", ".jpg", ".jpeg", ".bmp", ".tiff"), ) -> Optional[str]: for image_type in image_types: path = f"{image_prefix}{image_type}"