Skip to content

Commit

Permalink
Renamed DefaultModelTranslatorBattery class to ModelTranslatorVisitor…
Browse files Browse the repository at this point in the history
…Battery
  • Loading branch information
vruusmann committed Mar 2, 2022
1 parent ff98199 commit 6196218
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,9 @@
import org.jpmml.evaluator.visitors.ElementOptimizerBattery;
import org.jpmml.model.visitors.VisitorBattery;

public class DefaultModelTranslatorBattery extends VisitorBattery {
public class ModelTranslatorVisitorBattery extends VisitorBattery {

public DefaultModelTranslatorBattery(){
public ModelTranslatorVisitorBattery(){
addAll(new AttributeOptimizerBattery());
addAll(new ElementOptimizerBattery());
}
Expand Down
4 changes: 2 additions & 2 deletions src/main/java/org/jpmml/transpiler/TranspilerUtil.java
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@
import org.jpmml.model.visitors.VisitorBattery;
import org.jpmml.translator.PMMLObjectUtil;
import org.jpmml.translator.TranslationContext;
import org.jpmml.translator.visitors.DefaultModelTranslatorBattery;
import org.jpmml.translator.visitors.ModelTranslatorVisitorBattery;

public class TranspilerUtil {

Expand All @@ -52,7 +52,7 @@ private TranspilerUtil(){

static
public JCodeModel translate(PMML pmml, String className){
VisitorBattery visitorBattery = new DefaultModelTranslatorBattery();
VisitorBattery visitorBattery = new ModelTranslatorVisitorBattery();

visitorBattery.applyTo(pmml);

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@
import org.jpmml.evaluator.ResultField;
import org.jpmml.evaluator.testing.SimpleArchiveBatch;
import org.jpmml.model.SerializationUtil;
import org.jpmml.translator.visitors.DefaultModelTranslatorBattery;
import org.jpmml.translator.visitors.ModelTranslatorVisitorBattery;
import org.jpmml.transpiler.InMemoryTranspiler;
import org.jpmml.transpiler.Transpiler;
import org.jpmml.transpiler.TranspilerTransformer;
Expand Down Expand Up @@ -82,7 +82,7 @@ public EvaluatorBuilder getEvaluatorBuilder() throws Exception {
.setDerivedFieldGuard(new FieldNameSet(8))
.setFunctionGuard(new FunctionNameStack(4));

evaluatorBuilder.setVisitors(new DefaultModelTranslatorBattery());
evaluatorBuilder.setVisitors(new ModelTranslatorVisitorBattery());

try(InputStream is = open(getPmmlPath())){
evaluatorBuilder.load(is);
Expand Down

0 comments on commit 6196218

Please sign in to comment.