diff --git a/toolkit/dataloader_mixins.py b/toolkit/dataloader_mixins.py index 946c3d27..87e0679b 100644 --- a/toolkit/dataloader_mixins.py +++ b/toolkit/dataloader_mixins.py @@ -752,8 +752,7 @@ class ClipImageFileItemDTOMixin: if len(img_files) > 1: img_files.remove(self.path) # randomly grab one - self.clip_image_path = random.choice(img_files) - return self.path + return random.choice(img_files) else: return self.clip_image_path