From 018bbee0e14e0fd0bfd17db0c064fe307ca66ef5 Mon Sep 17 00:00:00 2001 From: Jesper Dramsch Date: Mon, 10 Jun 2024 15:21:00 +0200 Subject: [PATCH] docs: separated docstring --- src/anemoi/models/layers/conv.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/anemoi/models/layers/conv.py b/src/anemoi/models/layers/conv.py index f71683e5..1f7a445d 100644 --- a/src/anemoi/models/layers/conv.py +++ b/src/anemoi/models/layers/conv.py @@ -76,7 +76,9 @@ def aggregate(self, edges_new: Tensor, edge_index: Adj, dim_size: Optional[int] class GraphTransformerConv(MessagePassing): - """Message passing part of graph transformer operator from 'Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classification' + """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) """