From 5628c90d4560a98fc6a8ee3feb6d22d40c4d7df7 Mon Sep 17 00:00:00 2001 From: Simon Lang Date: Wed, 19 Jun 2024 17:00:30 +0100 Subject: [PATCH] fix doc string (#5) --- src/anemoi/models/layers/conv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/anemoi/models/layers/conv.py b/src/anemoi/models/layers/conv.py index 8e5049f..e486367 100644 --- a/src/anemoi/models/layers/conv.py +++ b/src/anemoi/models/layers/conv.py @@ -77,7 +77,7 @@ def aggregate(self, edges_new: Tensor, edge_index: Adj, dim_size: Optional[int] class GraphTransformerConv(MessagePassing): """Message passing part of graph transformer operator. - + Adapted from 'Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classification' (https://arxiv.org/abs/2009.03509) """