From 32118ba6afab8171b82f332fdc42e38a404e697d Mon Sep 17 00:00:00 2001
From: Tomaz Bratanic <bratanic.tomaz@gmail.com>
Date: Wed, 9 Jan 2019 00:09:11 +0100
Subject: [PATCH] Enable to use only topK parameter in similarity algos

---
 .../main/java/org/neo4j/graphalgo/similarity/CosineProc.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/algo/src/main/java/org/neo4j/graphalgo/similarity/CosineProc.java b/algo/src/main/java/org/neo4j/graphalgo/similarity/CosineProc.java
index d8d241758..e35aab544 100644
--- a/algo/src/main/java/org/neo4j/graphalgo/similarity/CosineProc.java
+++ b/algo/src/main/java/org/neo4j/graphalgo/similarity/CosineProc.java
@@ -68,7 +68,7 @@ public Stream<SimilaritySummaryResult> cosine(
         SimilarityComputer<WeightedInput> computer = similarityComputer(skipValue);
         Stream<SimilarityResult> stream = generateStream(configuration, inputs, similarityCutoff, topN, topK, computer);
 
-        boolean write = configuration.isWriteFlag(false) && similarityCutoff > 0.0;
+        boolean write = configuration.isWriteFlag(false) && (similarityCutoff > 0.0 || topK > 0);
         return writeAndAggregateResults(configuration, stream, inputs.length, write, "SIMILAR");
     }