Merge branch 'master' into sync-condition-2023Apr11
This commit is contained in:
@@ -12,7 +12,7 @@ steps:
|
||||
queue: pipeline-loader
|
||||
- wait
|
||||
- block: ":rocket: Run this test job"
|
||||
if: build.pull_request.repository.fork == true
|
||||
if: build.pull_request.id != null
|
||||
#### -------- CONTAINER BUILD --------
|
||||
- label: ":docker: Build containers"
|
||||
commands:
|
||||
|
||||
@@ -6,7 +6,7 @@ steps:
|
||||
queue: pipeline-loader
|
||||
- wait
|
||||
- block: ":rocket: Run this test job"
|
||||
if: build.pull_request.repository.fork == true
|
||||
if: build.pull_request.id != null
|
||||
#### -------- BUILD --------
|
||||
- label: ":windows: Build XGBoost for Windows with CUDA"
|
||||
command: "tests/buildkite/build-win64-gpu.ps1"
|
||||
|
||||
@@ -9,7 +9,7 @@ steps:
|
||||
queue: pipeline-loader
|
||||
- wait
|
||||
- block: ":rocket: Run this test job"
|
||||
if: build.pull_request.repository.fork == true
|
||||
if: build.pull_request.id != null
|
||||
#### -------- CONTAINER BUILD --------
|
||||
- label: ":docker: Build containers"
|
||||
commands:
|
||||
|
||||
@@ -464,6 +464,7 @@ class TestPySparkLocal:
|
||||
def test_regressor_basic(self, reg_data: RegData) -> None:
|
||||
regressor = SparkXGBRegressor(pred_contrib_col="pred_contribs")
|
||||
model = regressor.fit(reg_data.reg_df_train)
|
||||
assert regressor.uid == model.uid
|
||||
pred_result = model.transform(reg_data.reg_df_test).collect()
|
||||
for row in pred_result:
|
||||
np.testing.assert_equal(row.prediction, row.expected_prediction)
|
||||
|
||||
Reference in New Issue
Block a user