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

Commit d3f8898

Browse files
committed
apply changes
1 parent 7af1b33 commit d3f8898

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

scala-package/macros/src/main/scala/org/apache/mxnet/APIDocGenerator.scala

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,11 @@ private[mxnet] object APIDocGenerator{
5656
def absClassGen(FILE_PATH : String, isSymbol : Boolean) : String = {
5757
// scalastyle:off
5858
val absClassFunctions = getSymbolNDArrayMethods(isSymbol)
59+
// Defines Operators that should not generated
60+
val notGenerated = Set("Custom")
61+
// TODO: Add Filter to the same location in case of refactor
5962
val absFuncs = absClassFunctions.filterNot(_.name.startsWith("_"))
63+
.filterNot(ele => notGenerated.contains(ele.name))
6064
.map(absClassFunction => {
6165
val scalaDoc = generateAPIDocFromBackend(absClassFunction)
6266
val defBody = generateAPISignature(absClassFunction, isSymbol)

0 commit comments

Comments
 (0)