lint.

dmlc · Nov 14, 2024 · 764a660 · 764a660
1 parent 25c43c9
commit 764a660
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 11 deletions.
diff --git a/python-package/xgboost/dask/utils.py b/python-package/xgboost/dask/utils.py
@@ -54,14 +54,13 @@ def get_address_from_user(
         for k in dconfig:
             if k not in valid_config:
                 raise ValueError(f"Unknown configuration: {k}")
-            else:
-                warnings.warn(
-                    (
-                        "Use `coll_cfg` instead of the Dask global configuration store"
-                        f" for the XGBoost tracker configuration: {k}."
-                    ),
-                    FutureWarning,
-                )
+            warnings.warn(
+                (
+                    "Use `coll_cfg` instead of the Dask global configuration store"
+                    f" for the XGBoost tracker configuration: {k}."
+                ),
+                FutureWarning,
+            )
     else:
         dconfig = {}
 

diff --git a/python-package/xgboost/spark/core.py b/python-package/xgboost/spark/core.py
@@ -1043,9 +1043,13 @@ def _get_tracker_args(self) -> Tuple[bool, Dict[str, Any]]:
                 tracker_port = self.getOrDefault(self.tracker_port)
 
             num_workers = self.getOrDefault(self.num_workers)
-            rabit_args.update(_get_rabit_args(tracker_host_ip, num_workers, tracker_port))
+            rabit_args.update(
+                _get_rabit_args(tracker_host_ip, num_workers, tracker_port)
+            )
         else:
-            if self.isDefined(self.tracker_host_ip) or self.isDefined(self.tracker_port):
+            if self.isDefined(self.tracker_host_ip) or self.isDefined(
+                self.tracker_port
+            ):
                 raise ValueError(
                     "You must enable launch_tracker_on_driver to use "
                     "tracker_host_ip and tracker_port"

diff --git a/tests/test_distributed/test_with_spark/test_spark_local.py b/tests/test_distributed/test_with_spark/test_spark_local.py
@@ -1657,7 +1657,9 @@ def test_tracker(self):
             classifier._get_tracker_args()
 
         classifier = SparkXGBClassifier(
-            launch_tracker_on_driver=False, tracker_host_ip="127.0.0.1", tracker_port=58892
+            launch_tracker_on_driver=False,
+            tracker_host_ip="127.0.0.1",
+            tracker_port=58892,
         )
         with pytest.raises(
             ValueError, match="You must enable launch_tracker_on_driver"