Merge pull request #12 from jacekwasilewski/fix-10

Fix for issue #10
RankSys · Nov 21, 2015 · 043e0c6 · 043e0c6
2 parents 264c6bf + ebbb478
commit 043e0c6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/RankSys-examples/src/main/java/es/uam/eps/ir/ranksys/examples/RerankerExample.java b/RankSys-examples/src/main/java/es/uam/eps/ir/ranksys/examples/RerankerExample.java
@@ -64,12 +64,13 @@ public static void main(String[] args) throws Exception {
 
         RecommendationFormat<Long, Long> format = new SimpleRecommendationFormat<>(lp, lp);
 
-        rerankersMap.forEach((name, reranker) -> {
+        rerankersMap.forEach((name, rerankerSupplier) -> {
             System.out.println("Running " + name);
             String recOut = String.format("%s_%s", recIn, name);
+            Reranker<Long, Long> reranker = rerankerSupplier.get();
             try (RecommendationFormat.Writer<Long, Long> writer = format.getWriter(recOut)) {
                 format.getReader(recIn).readAll()
-                        .map(rec -> reranker.get().rerankRecommendation(rec, cutoff))
+                        .map(rec -> reranker.rerankRecommendation(rec, cutoff))
                         .forEach(rerankedRecommendation -> {
                             try {
                                 writer.write(rerankedRecommendation);