From 55a6a7f633532fbd13fd250a4c4b3e1f0a6e53e5 Mon Sep 17 00:00:00 2001
From: Priyesh Kakka <39474544+kakkapriyesh@users.noreply.github.com>
Date: Tue, 15 Aug 2023 02:27:46 -0400
Subject: [PATCH] Update utilities.py

This is required for l>1
---
 graph-neural-operator/utilities.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/graph-neural-operator/utilities.py b/graph-neural-operator/utilities.py
index af44e1a..b32ce81 100644
--- a/graph-neural-operator/utilities.py
+++ b/graph-neural-operator/utilities.py
@@ -493,7 +493,7 @@ def assemble(self, pred, split_idx, batch_size2, sigma=1):
             for j in range(batch_size2):
                 pred_ij = pred_i[j,:].reshape(-1,)
                 idx = split_idx_i[j,:].reshape(-1,)
-                out[idx] = pred_ij
+                out[idx] += pred_ij
 
         out = out / self.l