diff --git a/dataloader.py b/dataloader.py
index d45ab4db67e6e86d4e143fbe4382f45a7f4488d2..6f9f4f6bcc4d4d1929e0ce58e716f246a9fd2e03 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)