From e93a95f1b3a664392680d3a262d2ce1df2bd0bc4 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Tue, 26 Jul 2022 19:24:40 +0100
Subject: [PATCH] ai dataset: add if main == main

---
 aimodel/src/lib/dataset/dataset.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/aimodel/src/lib/dataset/dataset.py b/aimodel/src/lib/dataset/dataset.py
index b3c07f2..d5ca3c6 100644
--- a/aimodel/src/lib/dataset/dataset.py
+++ b/aimodel/src/lib/dataset/dataset.py
@@ -56,4 +56,7 @@ def dataset(dirpath_input, batch_size=64, train_percentage=0.8):
 
 
 if __name__ == "__main__":
-    dataset()
\ No newline at end of file
+	ds_train, ds_validate = dataset()
+	for thing in ds_validate():
+		as_str = str(thing)
+		print(thing[:100])
\ No newline at end of file