in src/model.py [0:0]
def forward(self, g: dgl.DGLGraph) -> torch.Tensor: h = self._conv_forward(g) h = self.gap(g, h) return self.output(h)