diff --git a/supervision/dataset/core.py b/supervision/dataset/core.py index 725d4ac9..9143164c 100644 --- a/supervision/dataset/core.py +++ b/supervision/dataset/core.py @@ -626,7 +626,9 @@ class ClassificationDataset(BaseDataset): else classification.get_top_k(1)[0][0] ) class_name = self.classes[class_id] - image_path = os.path.join(root_directory_path, class_name, f"{image_name}.{image_ext}") + image_path = os.path.join( + root_directory_path, class_name, f"{image_name}.{image_ext}" + ) cv2.imwrite(image_path, image) @classmethod