add dataview support in source generator

src/Microsoft.ML.AutoML/CodeGen/map_value_to_key_search_space.json

@@ -14,6 +14,10 @@
       "name": "AddKeyValueAnnotationsAsText",
       "type": "boolean",
       "default": false
+    },
+    {
+      "name": "KeyData",
+      "type": "dataView"
     }
   ]
 }

src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/MapValueToKey.cs

@@ -8,15 +8,15 @@ internal partial class MapValueToKey
     {
         public override IEstimator<ITransformer> BuildFromOption(MLContext context, MapValueToKeyOption param)
         {
-            return context.Transforms.Conversion.MapValueToKey(param.OutputColumnName, param.InputColumnName, addKeyValueAnnotationsAsText: param.AddKeyValueAnnotationsAsText);
+            return context.Transforms.Conversion.MapValueToKey(param.OutputColumnName, param.InputColumnName, addKeyValueAnnotationsAsText: param.AddKeyValueAnnotationsAsText, keyData: param.KeyData);
         }
     }
 
     internal partial class MapKeyToValue
     {
         public override IEstimator<ITransformer> BuildFromOption(MLContext context, MapKeyToValueOption param)
         {
-            return context.Transforms.Conversion.MapKeyToValue(param.OutputColumnName, param.InputColumnName, param.KeyData);
+            return context.Transforms.Conversion.MapKeyToValue(param.OutputColumnName, param.InputColumnName);
         }
     }
 }

tools-local/Microsoft.ML.AutoML.SourceGenerator/SearchSpaceGenerator.cs

@@ -57,6 +57,7 @@ public void Execute(GeneratorExecutionContext context)
                         "bertArchitecture" => "BertArchitecture",
                         "imageClassificationArchType" => "Microsoft.ML.Vision.ImageClassificationTrainer.Architecture",
                         "dataKind" => "Microsoft.ML.Data.DataKind",
+                        "dataView" => "Microsoft.ML.IDataView",
                         _ => throw new ArgumentException("unknown type"),
                     };
 
@@ -76,6 +77,7 @@ public void Execute(GeneratorExecutionContext context)
                         (_, "BertArchitecture") => defaultToken.GetValue<string>(),
                         (_, "Microsoft.ML.Vision.ImageClassificationTrainer.Architecture") => defaultToken.GetValue<string>(),
                         (_, "Microsoft.ML.Data.DataKind") => defaultToken.GetValue<string>(),
+                        (_, "Microsoft.ML.IDataView") => defaultToken.GetValue<string>(),
                         (_, _) => throw new ArgumentException("unknown"),
                     };