3e76bcc0528e2cd1584837162b3f286048fcd7ea,tests/pytorch/test_basics.py,,test_pull_0deg,#,224

Before Change

    old_repr = th.randn(2, 5)
    g.set_n_repr(old_repr)
    g.pull([0, 1], _message, _reduce)
    new_repr = g.get_n_repr()
    assert th.allclose(new_repr[0], old_repr[0])
    assert th.allclose(new_repr[1], old_repr[0])

def _disabled_test_send_twice():

After Change

    old_repr = th.randn(2, 5)
    g.set_n_repr({"h" : old_repr})
    g.pull([0, 1], _message, _reduce)
    new_repr = g.get_n_repr()["h"]
    assert th.allclose(new_repr[0], old_repr[0])
    assert th.allclose(new_repr[1], old_repr[0])

def _disabled_test_send_twice():

Instances