def forward()

in model/layers.py [0:0]

• 17 lines of code
• 1 McCabe index (conditional complexity)

    def forward(self, p, edge_index):
        net = self.fc_pos(p)
        net = self.block_0(net, edge_index)

        pooled = self.pool(net, dim=1, keepdim=True).expand(net.size())
        net = torch.cat([net, pooled, p], dim=1)

        net = self.block_1(net, edge_index)
        pooled = self.pool(net, dim=1, keepdim=True).expand(net.size())
        net = torch.cat([net, pooled, p], dim=1)

        net = self.block_2(net, edge_index)
        pooled = self.pool(net, dim=1, keepdim=True).expand(net.size())
        net = torch.cat([net, pooled, p], dim=1)

        net = self.block_3(net, edge_index)
        pooled = self.pool(net, dim=1, keepdim=True).expand(net.size())
        net = torch.cat([net, pooled, p], dim=1)

        net = self.block_4(net, edge_index)

        c = self.fc_c(self.actvn(net))

        return c