Improved the structuring of Java code

jpmml · Nov 29, 2022 · 792fd5d · 792fd5d
1 parent 181f5c7
commit 792fd5d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pmml-transpiler/src/main/java/org/jpmml/translator/NormContinuousTranslator.java b/pmml-transpiler/src/main/java/org/jpmml/translator/NormContinuousTranslator.java
@@ -43,6 +43,7 @@
 import org.dmg.pmml.NormContinuous;
 import org.dmg.pmml.OpType;
 import org.dmg.pmml.OutlierTreatmentMethod;
+import org.dmg.pmml.PMML;
 import org.jpmml.evaluator.FieldValue;
 import org.jpmml.evaluator.FieldValueUtil;
 import org.jpmml.model.InvalidElementException;
@@ -160,7 +161,7 @@ public void translateExpression(TranslationContext context){
 	}
 
 	private JDefinedClass ensureNormContinuousFuncInterface(TranslationContext context){
-		JDefinedClass owner = context.getOwner();
+		JDefinedClass owner = context.getOwner(PMML.class);
 
 		JDefinedClass definedClazz = JCodeModelUtil.getNestedClass(owner, "NormContinuousFunction");
 		if(definedClazz != null){
@@ -185,7 +186,7 @@ private JDefinedClass ensureNormContinuousFuncInterface(TranslationContext conte
 	private JMethod ensureCreateNormContinuousFuncMethod(JDefinedClass normContinuousFuncInterface, TranslationContext context){
 		JCodeModel codeModel = context.getCodeModel();
 
-		JDefinedClass owner = context.getOwner();
+		JDefinedClass owner = context.getOwner(PMML.class);
 
 		JType doubleType = context._ref(double.class);