Merge pull request #13 from mnbjhu/rand

Fixing collect type
mnbjhu · Jan 28, 2023 · dc460da · dc460da
2 parents 3f8ca88 + 415b32c
commit dc460da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/kotlin/uk/gibby/neo4k/functions/array_return/Functions.kt b/src/main/kotlin/uk/gibby/neo4k/functions/array_return/Functions.kt
@@ -41,4 +41,4 @@ fun range(start: LongReturn, end: Long): ArrayReturn<Long, LongReturn> = ReturnV
 fun range(start: Long, end: LongReturn): ArrayReturn<Long, LongReturn> = ReturnValue.createReference(intArrayType, "range($start,${end.getString()})") as ArrayReturn<Long, LongReturn>
 fun range(start: Long, end: Long): ArrayReturn<Long, LongReturn> = ReturnValue.createReference(intArrayType, "range($start,$end)") as ArrayReturn<Long, LongReturn>
 
-fun <T, U: DataType<T>>collect(expression: U) = array(TypeProducer(expression.createDummy())).inner.createReference("collect(${expression.getString()})")
+fun <T, U: DataType<T>>collect(expression: U): ArrayReturn<T, U> = array(TypeProducer(expression.createDummy())).inner.createReference("collect(${expression.getString()})") as ArrayReturn<T, U>