Updated JPMML-XGBoost dependency. Fixes jpmml/sklearn2pmml#402

jpmml · Dec 11, 2023 · 65c85ec · 65c85ec
1 parent ddfc8fe
commit 65c85ec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pmml-sklearn-xgboost/src/main/java/xgboost/sklearn/BoosterUtil.java b/pmml-sklearn-xgboost/src/main/java/xgboost/sklearn/BoosterUtil.java
@@ -92,6 +92,9 @@ public <E extends Estimator & HasBooster & HasXGBoostOptions> PMML encodePMML(E
 		Learner learner = getLearner(estimator);
 
 		FeatureMap featureMap = learner.encodeFeatureMap();
+		if(featureMap == null){
+			throw new IllegalArgumentException();
+		}
 
 		// XXX
 		Map<String, ?> options = estimator.getNativeConfiguration();

diff --git a/pom.xml b/pom.xml
@@ -129,7 +129,7 @@
 			<dependency>
 				<groupId>org.jpmml</groupId>
 				<artifactId>pmml-xgboost</artifactId>
-				<version>1.7.4</version>
+				<version>1.8.0</version>
 			</dependency>
 
 			<dependency>