Fixed test

ankane · ankane · commit 85bc90aa74be · 2025-10-18T12:53:18.000-07:00
diff --git a/lib/xgboost/model.rb b/lib/xgboost/model.rb
@@ -19,7 +19,7 @@ def save_model(fname)
     end
 
     def load_model(fname)
-      @booster = Booster.new(params: @params, model_file: fname)
+      @booster = Booster.new(model_file: fname)
     end
 
     def feature_importances
diff --git a/test/classifier_test.rb b/test/classifier_test.rb
@@ -44,8 +44,7 @@ def test_multiclass
 
     model = XGBoost::Classifier.new
     model.load_model(tempfile)
-    # TODO fix
-    # assert_equal y_pred, model.predict(x_test)
+    assert_equal y_pred, model.predict(x_test)
   end
 
   def test_early_stopping
