From 47eefe1861f1bc895b212c894bc54c9796aacd51 Mon Sep 17 00:00:00 2001 From: Abby Bertics Date: Tue, 10 Dec 2024 12:35:25 -0800 Subject: [PATCH] Make GNNWrapper forward call to backbone compatible with **kwargs --- topobenchmark/nn/wrappers/graph/gnn_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/topobenchmark/nn/wrappers/graph/gnn_wrapper.py b/topobenchmark/nn/wrappers/graph/gnn_wrapper.py index fd479281..c0576823 100644 --- a/topobenchmark/nn/wrappers/graph/gnn_wrapper.py +++ b/topobenchmark/nn/wrappers/graph/gnn_wrapper.py @@ -27,7 +27,7 @@ def forward(self, batch): x_0 = self.backbone( batch.x_0, batch.edge_index, - batch.get("edge_weight", None), + edge_weight=batch.get("edge_weight", None), ) model_out = {"labels": batch.y, "batch_0": batch.batch_0}