change type (intel-analytics#2959)

dding3 · Oct 16, 2020 · f90e0aa · f90e0aa
1 parent 77b12d9
commit f90e0aa
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/spark/dl/src/main/scala/com/intel/analytics/bigdl/dllib/inference/PythonInferenceModel.scala b/spark/dl/src/main/scala/com/intel/analytics/bigdl/dllib/inference/PythonInferenceModel.scala
@@ -81,12 +81,13 @@ class PythonInferenceModel[T: ClassTag](implicit ev: TensorNumeric[T]) extends P
       model: InferenceModel,
       modelPath: String,
       modelType: String,
-      inputs: Array[String],
-      outputs: Array[String],
+      inputs: JList[String],
+      outputs: JList[String],
       intraOpParallelismThreads: Int,
       interOpParallelismThreads: Int,
       usePerSessionThreads: Boolean): Unit = {
-    model.doLoadTensorflow(modelPath, modelType, inputs, outputs, intraOpParallelismThreads,
+    model.doLoadTensorflow(modelPath, modelType, Option(inputs).map(_.asScala.toArray).orNull,
+      Option(outputs).map(_.asScala.toArray).orNull, intraOpParallelismThreads,
       interOpParallelismThreads, usePerSessionThreads)
   }