lib/xgboost/dmatrix.rb in xgb-0.2.1 vs lib/xgboost/dmatrix.rb in xgb-0.3.0

@@ -58,10 +58,10 @@
     end
 
     def group=(group)
       c_data = ::FFI::MemoryPointer.new(:int, group.size)
       c_data.write_array_of_int(group)
-      check_result FFI.XGDMatrixSetGroup(handle_pointer, c_data, group.size)
+      check_result FFI.XGDMatrixSetUIntInfo(handle_pointer, "group", c_data, group.size)
     end
 
     def num_row
       out = ::FFI::MemoryPointer.new(:uint64)
       check_result FFI.XGDMatrixNumRow(handle_pointer, out)