Fix SUM over block floats (result should be float, not double)

1 files changed, 2 insertions, 2 deletions

diff --git a/ydb/library/yql/minikql/comp_nodes/mkql_block_agg_sum.cpp b/ydb/library/yql/minikql/comp_nodes/mkql_block_agg_sum.cpp
index e91762c2b6..7a663a2f45 100644
--- a/ydb/library/yql/minikql/comp_nodes/mkql_block_agg_sum.cpp
+++ b/ydb/library/yql/minikql/comp_nodes/mkql_block_agg_sum.cpp
@@ -623,7 +623,7 @@ std::unique_ptr<typename TTag::TPreparedAggregator> PrepareSum(TTupleType* tuple
         sumRetType = TDataType::Create(NUdf::TDataType<ui64>::Id, env);
     } else {
         Y_ENSURE(typeInfo.Features & NYql::NUdf::EDataTypeFeatures::FloatType);
-        sumRetType = TDataType::Create(NUdf::TDataType<double>::Id, env);
+        sumRetType = dataType;
     }
     sumRetType = TOptionalType::Create(sumRetType, env);
 
@@ -646,7 +646,7 @@ std::unique_ptr<typename TTag::TPreparedAggregator> PrepareSum(TTupleType* tuple
     case NUdf::EDataSlot::Uint64:
         return PrepareSumFixed<TTag, ui64, ui64>(sumRetType, isOptional, isScalar, filterColumn, argColumn);
     case NUdf::EDataSlot::Float:
-        return PrepareSumFixed<TTag, float, double>(sumRetType, isOptional, isScalar, filterColumn, argColumn);
+        return PrepareSumFixed<TTag, float, float>(sumRetType, isOptional, isScalar, filterColumn, argColumn);
     case NUdf::EDataSlot::Double:
         return PrepareSumFixed<TTag, double, double>(sumRetType, isOptional, isScalar, filterColumn, argColumn);
     default: