Fixed type for out_len

ankane · Oct 17, 2024 · e99a9ce · e99a9ce
1 parent 71be1f8
commit e99a9ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/xgboost/dmatrix.rb b/lib/xgboost/dmatrix.rb
@@ -231,7 +231,7 @@ def set_float_info(field, data)
 
     def float_info(field)
       num_row ||= num_row()
-      out_len = ::FFI::MemoryPointer.new(:int)
+      out_len = ::FFI::MemoryPointer.new(:uint64)
       out_dptr = ::FFI::MemoryPointer.new(:float, num_row)
       check_call FFI.XGDMatrixGetFloatInfo(handle, field, out_len, out_dptr)
       out_dptr.read_pointer.read_array_of_float(num_row)