Use black on more Python files. (#8137)

This commit is contained in:
Jiaming Yuan
2022-08-11 01:38:11 +08:00
committed by GitHub
parent bdb291f1c2
commit 570f8ae4ba
14 changed files with 183 additions and 133 deletions

View File

@@ -1,11 +1,10 @@
import sys
import logging
import random
import sys
import uuid
import numpy as np
import pytest
import testing as tm
if tm.no_spark()["condition"]:
@@ -13,26 +12,27 @@ if tm.no_spark()["condition"]:
if sys.platform.startswith("win") or sys.platform.startswith("darwin"):
pytest.skip("Skipping PySpark tests on Windows", allow_module_level=True)
from pyspark.ml.functions import vector_to_array
from pyspark.sql import functions as spark_sql_func
from pyspark.ml import Pipeline, PipelineModel
from pyspark.ml.evaluation import (
BinaryClassificationEvaluator,
MulticlassClassificationEvaluator,
)
from pyspark.ml.functions import vector_to_array
from pyspark.ml.linalg import Vectors
from pyspark.ml.tuning import CrossValidator, ParamGridBuilder
from pyspark.sql import functions as spark_sql_func
from xgboost.spark import (
SparkXGBClassifier,
SparkXGBClassifierModel,
SparkXGBRegressor,
SparkXGBRegressorModel,
)
from .utils import SparkTestCase
from xgboost import XGBClassifier, XGBRegressor
from xgboost.spark.core import _non_booster_params
from xgboost import XGBClassifier, XGBRegressor
from .utils import SparkTestCase
logging.getLogger("py4j").setLevel(logging.INFO)