set seed to avoid random error in PredictionServiceUT (intel-analytic…

…s#3097)
dding3 · Dec 10, 2020 · b19a259 · b19a259
1 parent fb2da24
commit b19a259
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/spark/dl/src/test/scala/com/intel/analytics/bigdl/optim/PredictionServiceSpec.scala b/spark/dl/src/test/scala/com/intel/analytics/bigdl/optim/PredictionServiceSpec.scala
@@ -18,6 +18,7 @@ package com.intel.analytics.bigdl.optim
 
 import com.intel.analytics.bigdl.nn._
 import com.intel.analytics.bigdl.tensor.Tensor
+import com.intel.analytics.bigdl.utils.RandomGenerator.RNG
 import com.intel.analytics.bigdl.utils.T
 import org.scalatest.{FlatSpec, Matchers}
 
@@ -134,6 +135,8 @@ class PredictionServiceSpec extends FlatSpec with Matchers {
     val service = PredictionService[Float](testModule, 4)
     val service2 = PredictionService[Float](testModule, 1)
 
+    RNG.setSeed(100)
+
     val sumResults = (1 to 100).par.map { _ =>
       val tensor = Tensor[Float](2, 10).randn()
       val output = service.predict(tensor).asInstanceOf[Tensor[Float]]