Use dtype constant for predict [skip ci]

ankane · Dec 16, 2024 · a15c129 · a15c129
1 parent 4af0fe2
commit a15c129
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/lightgbm/inner_predictor.rb b/lib/lightgbm/inner_predictor.rb
@@ -105,7 +105,7 @@ def inner_predict_array2d(input, start_iteration, num_iteration, predict_type)
 
       out_num_preds = ::FFI::MemoryPointer.new(:int64)
       out_result = ::FFI::MemoryPointer.new(:double, n_preds)
-      safe_call FFI.LGBM_BoosterPredictForMat(@handle, data, 1, input.count, input.first.count, 1, predict_type, start_iteration, num_iteration, @pred_parameter, out_num_preds, out_result)
+      safe_call FFI.LGBM_BoosterPredictForMat(@handle, data, FFI::C_API_DTYPE_FLOAT64, input.count, input.first.count, 1, predict_type, start_iteration, num_iteration, @pred_parameter, out_num_preds, out_result)
       if n_preds != out_num_preds.read_int64
         raise Error, "Wrong length for predict results"
       end