From 8f84a315becd0ceddc40762ba606f303a5a85d9a Mon Sep 17 00:00:00 2001
From: "P. Sai Vinay" <33659563+V1NAY8@users.noreply.github.com>
Date: Sat, 7 Aug 2021 02:29:48 +0530
Subject: [PATCH] Add test case for pseudohubererror for XGBoost

---
 tests/ml/test_ml_model_pytest.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tests/ml/test_ml_model_pytest.py b/tests/ml/test_ml_model_pytest.py
index e5dd1e6..2c10bd1 100644
--- a/tests/ml/test_ml_model_pytest.py
+++ b/tests/ml/test_ml_model_pytest.py
@@ -336,7 +336,13 @@ class TestMLModel:
     @pytest.mark.parametrize("compress_model_definition", [True, False])
     @pytest.mark.parametrize(
         "objective",
-        ["reg:squarederror", "reg:squaredlogerror", "reg:linear", "reg:logistic"],
+        [
+            "reg:squarederror",
+            "reg:squaredlogerror",
+            "reg:linear",
+            "reg:logistic",
+            "reg:pseudohubererror",
+        ],
     )
     @pytest.mark.parametrize("booster", ["gbtree", "dart"])
     def test_xgb_regressor(self, compress_model_definition, objective, booster):