Merge branch 'master' into upgrade-spark-3.3

Author: WeichenXu123

mleap-core/src/main/scala/ml/combust/mleap/core/feature/MathUnaryModel.scala

@@ -35,8 +35,17 @@ object UnaryOperation {
   case object Logit extends UnaryOperation {
     override def name: String = "logit"
   }
+  case object Floor extends UnaryOperation {
+    override def name: String = "floor"
+  }
+  case object Ceil extends UnaryOperation {
+    override def name: String = "ceil"
+  }
+  case object Round extends UnaryOperation {
+    override def name: String = "round"
+  }
 
-  val all = Set(Log, Exp, Sqrt, Sin, Cos, Tan, Abs, Logit)
+  val all = Set(Log, Exp, Sqrt, Sin, Cos, Tan, Abs, Logit, Floor, Ceil, Round)
   val forName: Map[String, UnaryOperation] = all.map(o => (o.name, o)).toMap
 }
 
@@ -52,6 +61,9 @@ case class MathUnaryModel(operation: UnaryOperation) extends Model {
     case Tan => Math.tan(a)
     case Abs => Math.abs(a)
     case Logit => LogitHelper.logit(a)
+    case Floor => Math.floor(a)
+    case Ceil => Math.ceil(a)
+    case Round => Math.round(a)
     case _ => throw new RuntimeException(s"unsupported unary operation: $operation")
   }
 

mleap-core/src/test/scala/ml/combust/mleap/core/feature/MathUnaryModelSpec.scala

@@ -33,4 +33,7 @@ class MathUnaryModelSpec extends FunSpec {
   unaryLike(Tan, "tan", 88777.777, Math.tan(88777.777))
   unaryLike(Abs, "abs", -88777.777, Math.abs(-88777.777))
   unaryLike(Logit, "logit", 0.9, LogitHelper.logit(0.9))
+  unaryLike(Floor, "floor", 3.4, Math.floor(3.4))
+  unaryLike(Ceil, "ceil", 4.6, Math.ceil(4.6))
+  unaryLike(Round, "round", 8.9, Math.round(8.9))
 }