Skip to content

Commit d8332e1

Browse files
committed
Add dummy signature for cosine_similarity on Array(Real)
1 parent 9652426 commit d8332e1

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

presto-main-base/src/main/java/com/facebook/presto/operator/scalar/MathFunctions.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1651,7 +1651,7 @@ public static Double mapCosineSimilarity(@SqlType("map(varchar,double)") Block l
16511651
@ScalarFunction("cosine_similarity")
16521652
@SqlNullable
16531653
@SqlType(StandardTypes.DOUBLE)
1654-
public static Double arrayCosineSimilarity(@SqlType("array(double)") Block leftArray, @SqlType("array(double)") Block rightArray)
1654+
public static Double arrayCosineSimilarityDouble(@SqlType("array(double)") Block leftArray, @SqlType("array(double)") Block rightArray)
16551655
{
16561656
checkCondition(
16571657
leftArray.getPositionCount() == rightArray.getPositionCount(),
@@ -1675,6 +1675,16 @@ public static Double arrayCosineSimilarity(@SqlType("array(double)") Block leftA
16751675
return dotProduct / (normLeftArray * normRightArray);
16761676
}
16771677

1678+
@Description("cosine similarity between the given identical sized vectors represented as arrays")
1679+
@ScalarFunction("cosine_similarity")
1680+
@SqlNullable
1681+
@SqlType(StandardTypes.REAL)
1682+
public static Long arrayCosineSimilarityReal(@SqlType("array(real)") Block leftArray, @SqlType("array(real)") Block rightArray)
1683+
{
1684+
throw new PrestoException(INVALID_FUNCTION_ARGUMENT,
1685+
"This is a dummy Java UDF signature. The actual implementation is in C++ and available in Prestissimo.");
1686+
}
1687+
16781688
@Description("squared Euclidean distance between the given identical sized vectors represented as arrays")
16791689
@ScalarFunction("l2_squared")
16801690
@SqlType(StandardTypes.REAL)

0 commit comments

Comments
 (0)