Minor cleanup in ScoreAggregator

vitrivr · Dec 11, 2023 · c1e27bd · c1e27bd
1 parent 01e9f9d
commit c1e27bd
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/vitrivr-engine-query/src/main/kotlin/org/vitrivr/engine/query/transform/ScoreAggregator.kt b/vitrivr-engine-query/src/main/kotlin/org/vitrivr/engine/query/transform/ScoreAggregator.kt
@@ -21,9 +21,9 @@ class ScoreAggregator(
 
     override fun toFlow(scope: CoroutineScope): Flow<Retrieved.RetrievedWithScore> =
         input.toFlow(scope).map { retrieved ->
-            when {
-                retrieved is Retrieved.RetrievedWithScore -> retrieved//pass through
-                retrieved is Retrieved.RetrievedWithRelationship -> { //aggregate
+            when (retrieved) {
+                is Retrieved.RetrievedWithScore -> retrieved//pass through
+                is Retrieved.RetrievedWithRelationship -> { //aggregate
 
                     val scores =
                         retrieved.relationships.filter { rel -> rel.pred in this.relationshps && rel.obj.first == retrieved.id }
@@ -44,7 +44,6 @@ class ScoreAggregator(
                 }
 
                 else -> Retrieved.PlusScore(retrieved, 0f)
-
             }
         }