From 2583d2f01f32aaac53440b05ad11eb0a111ad002 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Fri, 21 Feb 2025 21:05:48 +0100
Subject: [PATCH] bugfix

---
 src/aiunn/finetune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py
index 3e0441f..e3bdf46 100644
--- a/src/aiunn/finetune.py
+++ b/src/aiunn/finetune.py
@@ -16,7 +16,7 @@ class UpscaleDataset(Dataset):
         for parquet_file in parquet_files:
             df = pd.read_parquet(parquet_file, columns=['image_512', 'image_1024']).head(10000)
             combined_df = pd.concat([combined_df, df], ignore_index=True)
-
+        self.df = combined_df
         self.transform = transform
 
     def __len__(self):