Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
fix CI error - codestyle
Browse files Browse the repository at this point in the history
  • Loading branch information
mdespriee committed Oct 30, 2018
1 parent 0a8d06b commit af2d29a
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 19 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -216,19 +216,17 @@ private[mxnet] trait RandomHelpers {

// unify call targets (random_xyz and sample_xyz) and unify their argument types
private def unifyRandom(func: Func, isSymbol: Boolean): Func = {
var typeConv = if (isSymbol)
var typeConv = if (isSymbol) {
Map(
"org.apache.mxnet.Symbol" -> "Any",
"org.apache.mxnet.Base.MXFloat" -> "Any",
"Int" -> "Any"
)
else
"Int" -> "Any")
} else {
Map(
"org.apache.mxnet.NDArray" -> "Any",
"org.apache.mxnet.Base.MXFloat" -> "Any",
"Int" -> "Any"
)

"Int" -> "Any")
}
func.copy(
name = func.name.replaceAll("(random|sample)_", ""),
listOfArgs = func.listOfArgs
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -95,9 +95,9 @@ private[mxnet] object TypedNDArrayAPIMacro extends TypedAPIMacroBase {
structGeneration(c)(functionDefs, annottees: _*)
}

override def apiReturnType = "org.apache.mxnet.NDArrayFuncReturn"
override def apiReturnType: String = "org.apache.mxnet.NDArrayFuncReturn"

override def apiDataType = "org.apache.mxnet.NDArray"
override def apiDataType: String = "org.apache.mxnet.NDArray"

override protected def buildArgDef(func: Func): List[String] = {
super.buildArgDef(func) ++ List(
Expand Down Expand Up @@ -155,9 +155,9 @@ private[mxnet] object TypedNDArrayRandomAPIMacro extends TypedAPIMacroBase
)
}

override def apiDataType = "Any"
override def apiDataType: String = "Any"

override def apiReturnType = "org.apache.mxnet.NDArrayFuncReturn"
override def apiReturnType: String = "org.apache.mxnet.NDArrayFuncReturn"

override def prepareBackendCall(function: Func): List[String] = {
val impl = ListBuffer[String]()
Expand Down Expand Up @@ -186,10 +186,11 @@ private[mxnet] object TypedNDArrayRandomAPIMacro extends TypedAPIMacroBase
""".stripMargin
}
} else {
if (arg.isOptional)
if (arg.isOptional) {
s"""if (!${arg.safeArgName}.isEmpty) map("${arg.argName}") = ${arg.safeArgName}.get"""
else
} else {
s"""map("${arg.argName}") = ${arg.safeArgName}"""
}
}
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -90,9 +90,9 @@ private[mxnet] object TypedSymbolAPIMacro extends TypedAPIMacroBase {
structGeneration(c)(functionDefs, annottees: _*)
}

override def apiReturnType = "org.apache.mxnet.Symbol"
override def apiReturnType: String = "org.apache.mxnet.Symbol"

override def apiDataType = "org.apache.mxnet.Symbol"
override def apiDataType: String = "org.apache.mxnet.Symbol"

override protected def buildArgDef(func: Func): List[String] = {
super.buildArgDef(func) ++ List(
Expand Down Expand Up @@ -142,9 +142,9 @@ private[mxnet] object TypedSymbolRandomAPIMacro extends TypedAPIMacroBase
structGeneration(c)(functionDefs, annottees: _*)
}

override def apiDataType = "Any"
override def apiDataType: String = "Any"

override def apiReturnType = "org.apache.mxnet.Symbol"
override def apiReturnType: String = "org.apache.mxnet.Symbol"

override protected def buildArgDef(func: Func): List[String] = {
super.buildArgDef(func) ++ List(
Expand All @@ -163,10 +163,11 @@ private[mxnet] object TypedSymbolRandomAPIMacro extends TypedAPIMacroBase
s"args = ${arg.safeArgName}.toSeq"
} else {
// all go in kwargs
if (arg.isOptional)
if (arg.isOptional) {
s"""if (!${arg.safeArgName}.isEmpty) map("${arg.argName}") = ${arg.safeArgName}.get"""
else
} else {
s"""map("${arg.argName}") = ${arg.safeArgName}"""
}
}
}

Expand Down

0 comments on commit af2d29a

Please sign in to comment.