Fix GPU L1 error. (#8749)

2023-02-04 03:02:00 +08:00
parent 16ef016ba7
commit 0e61ba57d6
6 changed files with 78 additions and 15 deletions
--- a/tests/test_distributed/test_with_spark/test_spark_local.py
+++ b/tests/test_distributed/test_with_spark/test_spark_local.py
@@ -215,7 +215,6 @@ MultiClfData = namedtuple("MultiClfData", ("multi_clf_df_train", "multi_clf_df_t

@pytest.fixture
 def multi_clf_data(spark: SparkSession) -> Generator[MultiClfData, None, None]:
-
    X = np.array([[1.0, 2.0, 3.0], [1.0, 2.0, 4.0], [0.0, 1.0, 5.5], [-1.0, -2.0, 1.0]])
    y = np.array([0, 0, 1, 2])
    cls1 = xgb.XGBClassifier()