From 3c86644c217e533100da6f331ff5222bc8258b67 Mon Sep 17 00:00:00 2001
From: Yandi <yandirzm@gmail.com>
Date: Sun, 22 Jan 2023 12:50:58 +0100
Subject: [PATCH] [Second try] Removed time, space data as they are useless
here
---
dataloader.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/dataloader.py b/dataloader.py
index d45ab4d..6f9f4f6 100644
--- a/dataloader.py
+++ b/dataloader.py
@@ -180,6 +180,8 @@ def get_test_dataloader(
)
return test_loader
+def transform_remove_space_time():
+ return lambda attributes : attributes[:,4:]
if __name__ == "__main__":
logging.basicConfig(filename='logs/dataloader_unit_test.log', level=logging.INFO)
--
GitLab