Relaxed method parameters

jpmml · Aug 4, 2024 · 2622ed2 · 2622ed2
1 parent 30e9065
commit 2622ed2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/GBDT.java b/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/GBDT.java
@@ -169,7 +169,7 @@ public void load(List<Section> sections){
 		}
 	}
 
-	public Schema encodeSchema(String targetName, List<String> targetCategories, LightGBMEncoder encoder){
+	public Schema encodeSchema(String targetName, List<String> targetCategories, ModelEncoder encoder){
 		ObjectiveFunction object_function_ = getObjectiveFunction();
 		if(object_function_ == null){
 			throw new IllegalStateException();