[Test] Enable kvstore test (#3079)

* try enable kvstore test

* fix

* fix

* seperate out kvstore test

* add comment

Jenkinsfile

@@ -57,7 +57,7 @@ def cpp_unit_test_win64() {
 def unit_test_linux(backend, dev) {
   init_git()
   unpack_lib("dgl-${dev}-linux", dgl_linux_libs)
-  timeout(time: 20, unit: 'MINUTES') {
+  timeout(time: 30, unit: 'MINUTES') {
     sh "bash tests/scripts/task_unit_test.sh ${backend} ${dev}"
   }
 }

tests/distributed/test_dist_graph_store.py

@@ -176,6 +176,7 @@ def check_dist_emb(g, num_clients, num_nodes, num_edges):
     # Test sparse emb
     try:
         emb = NodeEmbedding(g.number_of_nodes(), 1, 'emb1', emb_init)
+        nids = F.arange(0, int(g.number_of_nodes()))
         lr = 0.001
         optimizer = SparseAdagrad([emb], lr=lr)
         with F.record_grad():

tests/scripts/task_unit_test.sh

@@ -37,5 +37,7 @@ python3 -m pytest -v --junitxml=pytest_backend.xml tests/$DGLBACKEND || fail "ba
 
 export OMP_NUM_THREADS=1
 if [ $2 != "gpu" ]; then
-    python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml tests/distributed || fail "distributed"
+    python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml tests/distributed/*.py || fail "distributed"
+    # Seperate kvstore test to another process, to avoid hangs
+    python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml tests/distributed/kv_store/*.py || fail "distributed kvstore"
 fi