4f02bb751b312aec02eab82737382f1f94829a4a,examples/pytorch/dis_kvstore/client.py,,start_client,#,16

Before Change

                                      edata_partition_book=edata_partition_book)

    client.push(name="edata", id_tensor=ID[client.get_id()], data_tensor=th.tensor([[1.,1.,1.],[1.,1.,1.]]))
    client.push(name="ndata", id_tensor=ID[client.get_id()], data_tensor=th.tensor([[2.,2.,2.],[2.,2.,2.]]))

    client.barrier()

After Change

    tensor_edata = client.pull(name="edata", id_tensor=th.tensor([0,1,2,3,4,5,6,7]))
    tensor_ndata = client.pull(name="ndata", id_tensor=th.tensor([0,1,2,3,4,5,6,7]))

    print(tensor_edata)
    client.barrier()

Instances