fixing to_data_list

geometric-intelligence · May 14, 2024 · 7721ac9 · 7721ac9
1 parent ec42f1e
commit 7721ac9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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