disable dist kvstore tests

apache · Apr 22, 2020 · 7f83c3f · 7f83c3f
1 parent 6e6584f
commit 7f83c3f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/tests/nightly/dist_device_sync_kvstore.py b/tests/nightly/dist_device_sync_kvstore.py
@@ -127,5 +127,6 @@ def check_trainer_kv_update(update_on_kv):
 
 if __name__ == "__main__":
     test_sync_init()
-    test_sync_push_pull()
+    # TODO(szha): disabled due to repeated failures. tracked in #18098
+    # test_sync_push_pull()
     test_gluon_trainer_type()
diff --git a/tests/nightly/test_distributed_training-gpu.sh b/tests/nightly/test_distributed_training-gpu.sh
@@ -27,14 +27,14 @@ test_kvstore() {
         "-n 4 --launcher local python3 dist_device_sync_kvstore.py"
         "-n 4 --launcher local python3 dist_device_sync_kvstore_custom.py"
         "--p3 -n 4 --launcher local python3 dist_device_sync_kvstore_custom.py"
-        "-n 4 --launcher local python3 dist_sync_kvstore.py --type=init_gpu" 
+        "-n 4 --launcher local python3 dist_sync_kvstore.py --type=init_gpu"
     )
 
     for arg in "${test_args[@]}"; do
         python3 ../../tools/launch.py $arg
         if [ $? -ne 0 ]; then
             return $?
-        fi 
+        fi
     done
 }
 
@@ -50,4 +50,4 @@ test_horovod() {
 test_kvstore
 test_horovod
 
-exit $errors
+exit $errors