File tree Expand file tree Collapse file tree 1 file changed +8
-13
lines changed
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/aggregate Expand file tree Collapse file tree 1 file changed +8
-13
lines changed Original file line number Diff line number Diff line change @@ -155,17 +155,13 @@ case class CovSample(
155
155
156
156
override def eval (buffer : InternalRow ): Any = {
157
157
val count = buffer.getLong(countOffset)
158
- if (count > 0 ) {
159
- if (count > 1 ) {
160
- val Ck = buffer.getDouble(CkOffset )
161
- val cov = Ck / (count - 1 )
162
- if (cov.isNaN) {
163
- null
164
- } else {
165
- cov
166
- }
167
- } else {
158
+ if (count > 1 ) {
159
+ val Ck = buffer.getDouble(CkOffset )
160
+ val cov = Ck / (count - 1 )
161
+ if (cov.isNaN) {
168
162
null
163
+ } else {
164
+ cov
169
165
}
170
166
} else {
171
167
null
@@ -190,11 +186,10 @@ case class CovPopulation(
190
186
val count = buffer.getLong(countOffset)
191
187
if (count > 0 ) {
192
188
val Ck = buffer.getDouble(CkOffset )
193
- val cov = Ck / count
194
- if (cov.isNaN) {
189
+ if (Ck .isNaN) {
195
190
null
196
191
} else {
197
- cov
192
+ Ck / count
198
193
}
199
194
} else {
200
195
null
You can’t perform that action at this time.
0 commit comments