66d28b476f24f84a9b3a88ad0bf60a2a0c5bc720,test/nn/conv/test_agnn_conv.py,,test_agnn_conv,#,5

Before Change

    out = conv(x, edge_index)
    assert out.size() == (num_nodes, in_channels)

    jit_conv = conv.jittable(x=x, edge_index=edge_index)
    jit_conv = torch.jit.script(jit_conv)
    assert jit_conv(x, edge_index).tolist() == out.tolist()

    conv = AGNNConv(requires_grad=False)
    out = conv(x, edge_index)
    assert out.size() == (num_nodes, in_channels)

    jit_conv = conv.jittable(x=x, edge_index=edge_index)
    jit_conv = torch.jit.script(jit_conv)
    assert jit_conv(x, edge_index).tolist() == out.tolist()

After Change

    out = conv(x, edge_index)
    assert out.size() == (num_nodes, in_channels)

    jit = torch.jit.script(conv.jittable())
    assert jit(x, edge_index).tolist() == out.tolist()

Instances