fix: java params patch

microsoft · May 2, 2021 · 8a6a849 · 8a6a849
1 parent d7b86d3
commit 8a6a849
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/scala/com/microsoft/ml/spark/codegen/Wrappable.scala b/src/main/scala/com/microsoft/ml/spark/codegen/Wrappable.scala
@@ -260,6 +260,12 @@ trait PythonWrappable extends BaseWrappable {
             |    sql_ctx = SQLContext.getOrCreate(ctx)
             |    return DataFrame(self._java_obj.get$capName(), sql_ctx)
             |""".stripMargin
+      case _: TransformerParam | EstimatorParam | PipelineStageParam =>
+        s"""|
+            |def get$capName(self):
+            |${indent(docString, 1)}
+            |    return JavaParams._from_java(self._java_obj.get$capName())
+            |""".stripMargin
       case _ =>
         s"""|
             |def get$capName(self):