fix small issues

apache · Jun 21, 2018 · 167dacf · 167dacf
1 parent d3be857
commit 167dacf
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/scala-package/macros/src/main/scala/org/apache/mxnet/APIDocGenerator.scala b/scala-package/macros/src/main/scala/org/apache/mxnet/APIDocGenerator.scala
@@ -182,7 +182,7 @@ private[mxnet] object APIDocGenerator{
     val realName = if (aliasName == name.value) "" else s"(a.k.a., ${name.value})"
 
     val argList = argNames zip argTypes zip argDescs map { case ((argName, argType), argDesc) =>
-      val typeAndOption = CToScalaUtils.argumentCleaner(argType, argName, returnType)
+      val typeAndOption = CToScalaUtils.argumentCleaner(argName, argType, "org.apache.mxnet.Symbol")
       new absClassArg(argName, typeAndOption._1, argDesc, typeAndOption._2)
     }
     new absClassFunction(aliasName, desc.value, argList.toList, returnType)

diff --git a/scala-package/macros/src/main/scala/org/apache/mxnet/NDArrayMacro.scala b/scala-package/macros/src/main/scala/org/apache/mxnet/NDArrayMacro.scala
@@ -226,7 +226,7 @@ private[mxnet] object NDArrayMacro {
     // scalastyle:on println
     val argList = argNames zip argTypes map { case (argName, argType) =>
       val typeAndOption =
-        CToScalaUtils.argumentCleaner(argType, argName, "org.apache.mxnet.NDArray")
+        CToScalaUtils.argumentCleaner(argName, argType, "org.apache.mxnet.Symbol")
       new NDArrayArg(argName, typeAndOption._1, typeAndOption._2)
     }
     new NDArrayFunction(aliasName, argList.toList)

diff --git a/scala-package/macros/src/main/scala/org/apache/mxnet/SymbolMacro.scala b/scala-package/macros/src/main/scala/org/apache/mxnet/SymbolMacro.scala
@@ -231,7 +231,7 @@ private[mxnet] object SymbolImplMacros {
     // scalastyle:on println
     val argList = argNames zip argTypes map { case (argName, argType) =>
         val typeAndOption =
-          CToScalaUtils.argumentCleaner(argType, argName, "org.apache.mxnet.Symbol")
+          CToScalaUtils.argumentCleaner(argName, argType, "org.apache.mxnet.Symbol")
         new SymbolArg(argName, typeAndOption._1, typeAndOption._2)
     }
     new SymbolFunction(aliasName, argList.toList)