add model and solver that maybe works

This commit is contained in:
Filippo Olivo
2025-11-20 11:38:50 +01:00
parent d865556c9f
commit 31059bf86e
3 changed files with 120 additions and 50 deletions

View File

@@ -23,7 +23,6 @@ class FiniteDifferenceStep(MessagePassing):
"""
TODO: add docstring.
"""
# return self.message_net(x_j * edge_attr)
return x_j * edge_attr
def update(self, aggr_out, _):