Align device id in predict transform with predictor. (#6662)

2021-02-02 08:33:29 +08:00
parent d8ec7aad5a
commit a9ec0ea6da
6 changed files with 30 additions and 8 deletions
--- a/tests/cpp/objective/test_aft_obj.cc
+++ b/tests/cpp/objective/test_aft_obj.cc
@@ -49,7 +49,7 @@ static inline void CheckGPairOverGridPoints(
  const int num_point = 20;
  const double log_y_low = 1.0;
  const double log_y_high = 15.0;
-  
+
  obj->Configure({ {"aft_loss_distribution", dist_type},
                   {"aft_loss_distribution_scale", "1"} });