Merge branch 'master' into sync-condition-2023Apr11

This commit is contained in:
amdsc21
2023-04-17 18:33:00 +02:00
5 changed files with 9 additions and 4 deletions

View File

@@ -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:

View File

@@ -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"

View File

@@ -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:

View File

@@ -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)