Skip to content

Commit 37df289

Browse files
author
Mahmoud Hanafy
committed
Fix evaluate average UDF
1 parent bae160e commit 37df289

File tree

2 files changed

+2
-2
lines changed
  • src/main
    • java/com/highperformancespark/examples/dataframe
    • scala/com/high-performance-spark-examples/dataframe

2 files changed

+2
-2
lines changed

src/main/java/com/highperformancespark/examples/dataframe/JavaUDFs.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ public void merge(MutableAggregationBuffer buffer1, Row buffer2) {
6464

6565
@Override
6666
public Object evaluate(Row buffer) {
67-
return Math.pow(buffer.getDouble(1), 1.0 / buffer.getLong(0));
67+
return buffer.getDouble(1) / buffer.getLong(0);
6868
}
6969
}
7070

src/main/scala/com/high-performance-spark-examples/dataframe/UDFs.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ object UDFs {
4747
}
4848

4949
def evaluate(buffer: Row): Any = {
50-
math.pow(buffer.getDouble(1), 1.toDouble / buffer.getLong(0))
50+
buffer.getDouble(1) / buffer.getLong(0)
5151
}
5252
}
5353
// Optionally register

0 commit comments

Comments
 (0)