diff --git a/engines/ml/xgboost/src/test/java/ai/djl/ml/xgboost/XgbModelTest.java b/engines/ml/xgboost/src/test/java/ai/djl/ml/xgboost/XgbModelTest.java index 7d928b121f4..acbfa998867 100644 --- a/engines/ml/xgboost/src/test/java/ai/djl/ml/xgboost/XgbModelTest.java +++ b/engines/ml/xgboost/src/test/java/ai/djl/ml/xgboost/XgbModelTest.java @@ -53,7 +53,7 @@ public void downloadXGBoostModel() throws IOException { @Test public void testVersion() { Engine engine = Engine.getEngine("XGBoost"); - Assert.assertEquals("2.0.1", engine.getVersion()); + Assert.assertEquals("2.0.3", engine.getVersion()); } /* diff --git a/gradle.properties b/gradle.properties index b68cfce9c3f..9743ae28fd8 100644 --- a/gradle.properties +++ b/gradle.properties @@ -23,7 +23,7 @@ sentencepiece_version=0.1.97 tokenizers_version=0.15.0 llamacpp_version=b1696 fasttext_version=0.9.2 -xgboost_version=2.0.1 +xgboost_version=2.0.3 lightgbm_version=3.2.110 rapis_version=22.12.0