diff --git a/src/main/java/be/cylab/java/wowa/training/Factory.java b/src/main/java/be/cylab/java/wowa/training/Factory.java index 91e21d623bcd2d77e66ce4fc572915206605dec5..31877dada0e42369ca42936ec94a8149b34d2efc 100644 --- a/src/main/java/be/cylab/java/wowa/training/Factory.java +++ b/src/main/java/be/cylab/java/wowa/training/Factory.java @@ -1,6 +1,5 @@ package be.cylab.java.wowa.training; -import java.lang.reflect.InvocationTargetException; /** * Class to generate objects. @@ -18,13 +17,14 @@ public class Factory { /** * @return */ - public AbstractSolution createSolutionObject(final int a) { + public final AbstractSolution createSolutionObject(final int a) { if (this.solution_type instanceof SolutionDistance) { return new SolutionDistance(a); } else if (this.solution_type instanceof SolutionAUC) { return new SolutionAUC(a); } else { - throw new IllegalArgumentException("Provide o correct type of Solution"); + throw new IllegalArgumentException( + "Provide o correct type of Solution"); } } } diff --git a/src/main/java/be/cylab/java/wowa/training/Trainer.java b/src/main/java/be/cylab/java/wowa/training/Trainer.java index 194b89b2c038a2cb38a7f07fc8abfe0cf05633f7..ad22c51775c5cc713306d831c409afbb66fc1283 100644 --- a/src/main/java/be/cylab/java/wowa/training/Trainer.java +++ b/src/main/java/be/cylab/java/wowa/training/Trainer.java @@ -95,7 +95,8 @@ public class Trainer { List<AbstractSolution> population = new ArrayList<>(); for (int i = 0; i < population_size; i++) { - AbstractSolution solution = this.factory.createSolutionObject(number_of_weights); + AbstractSolution solution + = this.factory.createSolutionObject(number_of_weights); population.add(solution); } return population;