zachgk commented on a change in pull request #13159: [MXNET-1202] Change
Builder class into a better way
URL: https://github.com/apache/incubator-mxnet/pull/13159#discussion_r231685023
##########
File path:
scala-package/macros/src/main/scala/org/apache/mxnet/APIDocGenerator.scala
##########
@@ -203,26 +203,40 @@ private[mxnet] object APIDocGenerator{
}
def generateJavaAPISignature(func : absClassFunction) : String = {
+ val useParamObject = func.listOfArgs.count(arg => arg.isOptional) >= 2
var argDef = ListBuffer[String]()
var classDef = ListBuffer[String]()
func.listOfArgs.foreach(absClassArg => {
val currArgName = safetyNameCheck(absClassArg.argName)
// scalastyle:off
- if (absClassArg.isOptional) {
- classDef += s"def set${absClassArg.argName}(${absClassArg.argName} :
${absClassArg.argType}) : ${func.name}BuilderBase"
+ if (absClassArg.isOptional && useParamObject) {
+ classDef += s"def
set${absClassArg.argName.capitalize}(${absClassArg.argName}:
${absClassArg.argType}): ${func.name}ParamBase"
}
else {
argDef += s"$currArgName : ${absClassArg.argType}"
}
// scalastyle:on
})
- classDef += s"def setout(out : NDArray) : ${func.name}BuilderBase"
- classDef += s"def invoke() : org.apache.mxnet.javaapi.NDArrayFuncReturn"
val experimentalTag = "@Experimental"
- // scalastyle:off
- var finalStr = s"$experimentalTag\ndef ${func.name} (${argDef.mkString(",
")}) : ${func.name}BuilderBase\n"
- // scalastyle:on
- finalStr += s"abstract class ${func.name}BuilderBase {\n
${classDef.mkString("\n ")}\n}"
+ val returnType = "org.apache.mxnet.javaapi.NDArrayFuncReturn"
+ var finalStr = ""
Review comment:
You should try to combine all of these into as few string builders as
possible as well as reduce extra variables. Just throw the classDef and argDef
additions right into the string builder.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
