From 7721ac94442c88adf30efc0c522fe308dafeb486 Mon Sep 17 00:00:00 2001
From: levtelyatnikov <tlscabinet@gmail.com>
Date: Tue, 14 May 2024 20:19:18 +0200
Subject: [PATCH] fixing to_data_list

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

diff --git a/topobenchmarkx/data/dataloaders.py b/topobenchmarkx/data/dataloaders.py
index 0b0612b0..9fbc7863 100755
--- a/topobenchmarkx/data/dataloaders.py
+++ b/topobenchmarkx/data/dataloaders.py
@@ -108,7 +108,7 @@ def collate_fn(batch):
     # Ensure shape is torch.Tensor
     # "shape" describes the number of n_cells in each graph
     batch["shape"] = torch.Tensor(batch["shape"]).long()
-    to_data_list(batch)
+
     return batch