Unverified Commit 17f1432a authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[DistTest] fix incorrect shell if statement (#4304)

* [DistTest] fix incorrect shell if statement

* fix incorrect use of dist.initialize()
parent ee91863e
......@@ -84,7 +84,7 @@ def start_get_degrees_client(rank, tmpdir, disable_shared_mem, nids=None):
gpb = None
if disable_shared_mem:
_, _, _, gpb, _, _, _ = load_partition(tmpdir / 'test_get_degrees.json', rank)
dgl.distributed.initialize("rpc_ip_config.txt", 1)
dgl.distributed.initialize("rpc_ip_config.txt")
dist_graph = DistGraph("test_get_degrees", gpb=gpb)
try:
in_deg = dist_graph.in_degrees(nids)
......
......@@ -41,7 +41,7 @@ python3 -m pytest -v --junitxml=pytest_backend.xml --durations=100 tests/$DGLBAC
export PYTHONUNBUFFERED=1
export OMP_NUM_THREADS=1
export DMLC_LOG_DEBUG=1
if [ $2 != "gpu" && $DGLBACKEND == "pytorch" ]; then
if [ $2 != "gpu" ] && [ $DGLBACKEND == "pytorch" ]; then
python3 -m pip install filelock
python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml --durations=100 tests/distributed/*.py || fail "distributed"
PYTHONPATH=tools:$PYTHONPATH python3 -m pytest -v --capture=tee-sys --junitxml=pytest_tools.xml --durations=100 tests/tools/*.py || fail "tools"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment