From 82ee2317e80f93465685529e1ab9acff52b683c7 Mon Sep 17 00:00:00 2001
From: Liangcai Li <firestarmanllc@gmail.com>
Date: Wed, 25 Sep 2019 20:41:53 +0800
Subject: [PATCH] Add case for LongParam. (#4885)

To support specifying long parameter as String, the same as other basic
type, such as Int, Double ...
---
 .../ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala
index 8d6bc74ab..076dff42e 100644
--- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala
+++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/GeneralParams.scala
@@ -260,6 +260,8 @@ private[spark] trait ParamMapFuncs extends Params {
           set(name, paramValue.toString.toInt)
         case _: FloatParam =>
           set(name, paramValue.toString.toFloat)
+        case _: LongParam =>
+          set(name, paramValue.toString.toLong)
         case _: Param[_] =>
           set(name, paramValue)
       }