Remove port test.

dmlc · Nov 14, 2024 · c1f5137 · c1f5137
1 parent be0ddf5
commit c1f5137
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_distributed/test_with_spark/test_spark_local.py b/tests/test_distributed/test_with_spark/test_spark_local.py
@@ -4,7 +4,7 @@
 import tempfile
 import uuid
 from collections import namedtuple
-from typing import Generator, Sequence, Type
+from typing import Generator, Sequence
 
 import numpy as np
 import pytest
@@ -1667,11 +1667,10 @@ def test_tracker(self):
         classifier = SparkXGBClassifier(
             launch_tracker_on_driver=True,
             tracker_host="127.0.0.1",
-            tracker_port=58892,
             num_workers=2,
         )
         launch_tracker_on_driver, rabit_envs = classifier._get_tracker_args()
-        assert launch_tracker_on_driver == True
+        assert launch_tracker_on_driver is True
         assert rabit_envs["n_workers"] == 2
         assert rabit_envs["dmlc_tracker_uri"] == "127.0.0.1"