Add typehint to tracker. (#7338)

2021-10-20 12:49:36 +08:00
parent 5ff210ed75
commit f53da412aa
4 changed files with 122 additions and 94 deletions
--- a/tests/python/test_tracker.py
+++ b/tests/python/test_tracker.py
@@ -10,7 +10,7 @@ if sys.platform.startswith("win"):


 def test_rabit_tracker():
-    tracker = RabitTracker(hostIP='127.0.0.1', nslave=1)
+    tracker = RabitTracker(hostIP='127.0.0.1', n_workers=1)
    tracker.start(1)
    rabit_env = [
        str.encode('DMLC_TRACKER_URI=127.0.0.1'),