Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

correct loading files

This commit is contained in:
Falko Victor Habel 2025-01-26 17:26:13 +01:00
parent b7dc835a86
commit 9124221346
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/aiia/data/DataLoader.py
View File

@ -53,7 +53,7 @@ class JPGImageLoader:
self.successful_count = 0
self.skipped_count = 0
if self.bytes_column not in dataset.column_names:
if self.bytes_column not in dataset.columns:
raise ValueError(f"Column '{self.bytes_column}' not found in dataset.")
def _get_image(self, item):