use TypeCollection to specify wanted input and implicit cast

apache · Jul 14, 2015 · d10be4a · d10be4a
1 parent c3b9839
commit d10be4a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/math.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/math.scala
@@ -541,7 +541,10 @@ case class Round(child: Expression, scale: Expression) extends Expression with E
       case t => t
     }
 
-  override def inputTypes: Seq[AbstractDataType] = Seq(NumericType, IntegralType)
+  override def inputTypes: Seq[AbstractDataType] = Seq(
+    //rely on precedence to implicit cast String into Double
+    TypeCollection(DoubleType, FloatType, LongType, IntegerType, ShortType, ByteType),
+    TypeCollection(LongType, IntegerType, ShortType, ByteType))
 
   override def checkInputDataTypes(): TypeCheckResult = {
     child.dataType match {