Misc fixes

microsoft · Feb 23, 2018 · 4413771 · 4413771
1 parent e9f67ab
commit 4413771
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/codegen/src/main/scala/PySparkWrapperTest.scala b/src/codegen/src/main/scala/PySparkWrapperTest.scala
@@ -268,9 +268,10 @@ class PySparkTransformerWrapperTest(entryPoint: Transformer,
         case "ComputePerInstanceStatistics" => computeStatisticsString(entryPointName)
         case "IndexToValue" => indexToValueString(entryPointName)
         case "ValueIndexerModel" => valueIndexerModelString(entryPointName)
-        case "_CNTKModel" | "_UDFTransformer" | "FastVectorAssembler" | "MultiNGram" | "ImageFeaturizer"
-           | "_ImageFeaturizer" | "_ImageTransformer" | "UnrollImage" | "HashTransform" | "Timer"
-           | "StopWordsRemoverTransform"  | "ImageSetAugmenter"
+        case "_CNTKModel" | "FastVectorAssembler" | "_UDFTransformer"
+           | "MultiNGram" | "ImageFeaturizer" | "_ImageFeaturizer"
+           | "_ImageTransformer" | "UnrollImage" | "HashTransform" | "Timer"
+           | "StopWordsRemoverTransform"  | "ImageSetAugmenter" | "PowerBITransformer"
            => ""
         case _ =>
           tryFitSetupTemplate(entryPointName) + tryTransformString(entryPointName)