From c1f5137314d2f46ed53bc97e615710e43d5925e3 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Thu, 14 Nov 2024 13:32:07 +0800 Subject: [PATCH] Remove port test. --- tests/test_distributed/test_with_spark/test_spark_local.py | 5 ++--- 1 file changed, 2 insertions(+), 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 index 4521ec70d927..b88f94d64c61 100644 --- 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"