[Dist] enable partition tests (#7675)

Author: Rhett-Ying

tests/scripts/task_distributed_test.sh

@@ -34,8 +34,8 @@ export PYTHONUNBUFFERED=1
 export OMP_NUM_THREADS=1
 export DMLC_LOG_DEBUG=1
 
-# Tests for distributed are skipped due to glitch @2024.06.27.
-#python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml --durations=100 tests/distributed/*.py || fail "distributed"
+# Tests for distributed except test_partition.py are skipped due to glitch @2024.06.27.
+python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml --durations=100 tests/distributed/test_partition.py || fail "distributed"
 
 # Tests for tools are skipped due to glitch.
 #PYTHONPATH=tools:tools/distpartitioning:$PYTHONPATH python3 -m pytest -v --capture=tee-sys --junitxml=pytest_tools.xml --durations=100 tests/tools/*.py || fail "tools"