修正单元测试

songfang · Jun 25, 2018 · 7bfbff9 · 7bfbff9
1 parent 049e6db
commit 7bfbff9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/streamingpro-mlsql/src/main/java/streaming/dsl/mmlib/algs/SQLPythonAlg.scala b/streamingpro-mlsql/src/main/java/streaming/dsl/mmlib/algs/SQLPythonAlg.scala
@@ -199,7 +199,7 @@ class SQLPythonAlg extends SQLAlg with Functions {
     val pythonPath = metasTemp(0)("pythonPath")
     val pythonVer = metasTemp(0)("pythonVer")
 
-    val userPythonScript = findPythonPredictScript(sparkSession, params, "sk_predict.py")
+    val userPythonScript = findPythonPredictScript(sparkSession, params, "")
 
     val maps = new util.HashMap[String, java.util.Map[String, String]]()
     val item = new util.HashMap[String, String]()

diff --git a/streamingpro-mlsql/src/main/java/streaming/rest/RestController.scala b/streamingpro-mlsql/src/main/java/streaming/rest/RestController.scala
@@ -67,6 +67,7 @@ class RestController extends ApplicationController {
 
   @At(path = Array("/run/script"), types = Array(GET, POST))
   def script = {
+    restResponse.httpServletResponse().setHeader("Access-Control-Allow-Origin", "*")
     val silence = paramAsBoolean("silence", false)
     val sparkSession = runtime.asInstanceOf[SparkRuntime].sparkSession
     val htp = findService(classOf[HttpTransportService])

diff --git a/streamingpro-mlsql/src/main/resources-local/test/sql/python-alg-script.sql b/streamingpro-mlsql/src/main/resources-local/test/sql/python-alg-script.sql
@@ -16,7 +16,7 @@ and `systemParam.pythonVer`="2.7"
 ;
 
 register PythonAlg.`/tmp/pa_model` as jack options
-pythonScriptPPath="${pythonPredictScriptPath}"
+pythonScriptPath="${pythonPredictScriptPath}"
 and algIndex="0"
 ;