Employed null-safe getter methods

jpmml · Aug 2, 2024 · 95d05b5 · 95d05b5
1 parent 3679e74
commit 95d05b5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pmml-sklearn/src/main/java/sklearn/EstimatorUtil.java b/pmml-sklearn/src/main/java/sklearn/EstimatorUtil.java
@@ -153,7 +153,7 @@ public List<Feature> export(Estimator estimator, String predictFunc, Schema sche
 
 							OutputField outputField = Iterables.getLast(predictionOutputFields);
 
-							return encoder.export(model, outputField.getName());
+							return encoder.export(model, outputField.requireName());
 						} else
 
 						{

diff --git a/pmml-sklearn/src/main/java/sklearn/calibration/CalibratedClassifier.java b/pmml-sklearn/src/main/java/sklearn/calibration/CalibratedClassifier.java
@@ -140,7 +140,7 @@ public Model encodeModel(Schema schema){
 					}
 
 					// XXX
-					outputField.setName(getDecisionFunctionField(outputField.getName()));
+					outputField.setName(getDecisionFunctionField(outputField.requireName()));
 
 					models.add(decisionFunctionModel);
-Original file line number
+Diff line change
@@ Expand Up @@
     							OutputField outputField = Iterables.getLast(predictionOutputFields);
-    							return encoder.export(model, outputField.getName());
+    							return encoder.export(model, outputField.requireName());
     						} else
     						{
@@ Expand Down @@