Fix with black 24.1.1. (#10014)

python-package/xgboost/collective.py

@@ -1,4 +1,5 @@
 """XGBoost collective communication related API."""
+
 import ctypes
 import json
 import logging

python-package/xgboost/core.py

@@ -64,13 +64,11 @@ class XGBoostError(ValueError):
 
 
 @overload
-def from_pystr_to_cstr(data: str) -> bytes:
-    ...
+def from_pystr_to_cstr(data: str) -> bytes: ...
 
 
 @overload
-def from_pystr_to_cstr(data: List[str]) -> ctypes.Array:
-    ...
+def from_pystr_to_cstr(data: List[str]) -> ctypes.Array: ...
 
 
 def from_pystr_to_cstr(data: Union[str, List[str]]) -> Union[bytes, ctypes.Array]:

python-package/xgboost/dask/__init__.py

@@ -1146,9 +1146,9 @@ async def _direct_predict_impl(  # pylint: disable=too-many-branches
     if _can_output_df(isinstance(data, dd.DataFrame), output_shape):
         if base_margin is not None and isinstance(base_margin, da.Array):
             # Easier for map_partitions
-            base_margin_df: Optional[
-                Union[dd.DataFrame, dd.Series]
-            ] = base_margin.to_dask_dataframe()
+            base_margin_df: Optional[Union[dd.DataFrame, dd.Series]] = (
+                base_margin.to_dask_dataframe()
+            )
         else:
             base_margin_df = base_margin
         predictions = dd.map_partitions(

python-package/xgboost/dask/utils.py

@@ -1,4 +1,5 @@
 """Utilities for the XGBoost Dask interface."""
+
 import logging
 from typing import TYPE_CHECKING, Any, Dict
 

python-package/xgboost/spark/core.py

@@ -1,4 +1,5 @@
 """XGBoost pyspark integration submodule for core code."""
+
 import base64
 
 # pylint: disable=fixme, too-many-ancestors, protected-access, no-member, invalid-name

python-package/xgboost/spark/estimator.py

@@ -1,4 +1,5 @@
 """Xgboost pyspark integration submodule for estimator API."""
+
 # pylint: disable=too-many-ancestors
 # pylint: disable=fixme, too-many-ancestors, protected-access, no-member, invalid-name
 # pylint: disable=unused-argument, too-many-locals

python-package/xgboost/spark/params.py

@@ -1,4 +1,5 @@
 """Xgboost pyspark integration submodule for params."""
+
 from typing import Dict
 
 # pylint: disable=too-few-public-methods
@@ -55,7 +56,6 @@ class HasFeaturesCols(Params):
 
 
 class HasEnableSparseDataOptim(Params):
-
     """
     This is a Params based class that is extended by _SparkXGBParams
     and holds the variable to store the boolean config of enabling sparse data optimization.

python-package/xgboost/spark/utils.py

@@ -1,4 +1,5 @@
 """Xgboost pyspark integration submodule for helper functions."""
+
 # pylint: disable=fixme
 
 import inspect

python-package/xgboost/testing/__init__.py

@@ -2,6 +2,7 @@
 change without notice.
 
 """
+
 # pylint: disable=invalid-name,missing-function-docstring,import-error
 import gc
 import importlib.util

python-package/xgboost/testing/continuation.py

@@ -1,4 +1,5 @@
 """Tests for training continuation."""
+
 import json
 from typing import Any, Dict, TypeVar
 

python-package/xgboost/testing/dask.py

@@ -1,4 +1,5 @@
 """Tests for dask shared by different test modules."""
+
 import numpy as np
 import pandas as pd
 from dask import array as da

python-package/xgboost/testing/data_iter.py

@@ -1,4 +1,5 @@
 """Tests related to the `DataIter` interface."""
+
 import numpy as np
 
 import xgboost

python-package/xgboost/testing/metrics.py

@@ -1,4 +1,5 @@
 """Tests for evaluation metrics."""
+
 from typing import Dict, List
 
 import numpy as np

python-package/xgboost/testing/shared.py

@@ -1,4 +1,5 @@
 """Testing code shared by other tests."""
+
 # pylint: disable=invalid-name
 import collections
 import importlib.util

python-package/xgboost/testing/updater.py

@@ -1,4 +1,5 @@
 """Tests for updaters."""
+
 import json
 from functools import partial, update_wrapper
 from typing import Any, Dict, List