From 24cd94929e63566306b6f9c6afa41428664f657a Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Wed, 25 Jun 2025 14:01:08 -0600
Subject: [PATCH] Fix bug that can happen with fast processing dataset

---
 toolkit/data_transfer_object/data_loader.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/toolkit/data_transfer_object/data_loader.py b/toolkit/data_transfer_object/data_loader.py
index 0c7d7562..9078fbbb 100644
--- a/toolkit/data_transfer_object/data_loader.py
+++ b/toolkit/data_transfer_object/data_loader.py
@@ -91,6 +91,8 @@ class FileItemDTO(
                 except image_utils.UnknownImageFormat:
                     print_once(f'Warning: Some images in the dataset cannot be fast read. ' + \
                             f'This process is faster for png, jpeg')
+                    img = exif_transpose(Image.open(self.path))
+                    w, h = img.size
             else:
                 img = exif_transpose(Image.open(self.path))
                 w, h = img.size