From 46b1eda6d1c7db552ce5c577bed101c61f09d55b Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sat, 22 Oct 2022 16:56:10 +0200
Subject: Revert lat; fix skip attribute in dataset

---
 data/csv.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'data')

diff --git a/data/csv.py b/data/csv.py
index 4c91ded..df15c5a 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -76,7 +76,7 @@ class CSVDataModule(pl.LightningDataModule):
 
     def prepare_data(self):
         metadata = pd.read_json(self.data_file)
-        metadata = [item for item in metadata.itertuples() if "skip" not in item or item.skip != True]
+        metadata = [item for item in metadata.itertuples() if not hasattr(item, "skip") or item.skip != True]
         num_images = len(metadata)
 
         valid_set_size = int(num_images * 0.2)
-- 
cgit v1.2.3-54-g00ecf