From 9915e473e32726278fe89c6a77824d3d38ce487d Mon Sep 17 00:00:00 2001
From: XiaoYun Zhang <xiaoyuz@microsoft.com>
Date: Thu, 5 Jan 2023 13:56:28 -0800
Subject: [PATCH] add addKeyValueAnnotationsAsText flag

---
 .../CodeGen/map_value_to_key_search_space.json               | 5 +++++
 src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json     | 3 ++-
 .../SweepableEstimator/Estimators/MapValueToKey.cs           | 2 +-
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/Microsoft.ML.AutoML/CodeGen/map_value_to_key_search_space.json b/src/Microsoft.ML.AutoML/CodeGen/map_value_to_key_search_space.json
index 6b52fa5bb4..1663a95c18 100644
--- a/src/Microsoft.ML.AutoML/CodeGen/map_value_to_key_search_space.json
+++ b/src/Microsoft.ML.AutoML/CodeGen/map_value_to_key_search_space.json
@@ -9,6 +9,11 @@
     {
       "name": "InputColumnName",
       "type": "string"
+    },
+    {
+      "name": "AddKeyValueAnnotationsAsText",
+      "type": "boolean",
+      "default": false
     }
   ]
 }
diff --git a/src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json b/src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json
index 22432ebf80..935ff61643 100644
--- a/src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json
+++ b/src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json
@@ -177,7 +177,8 @@
         "Sentence2ColumnName",
         "BatchSize",
         "MaxEpochs",
-        "Architecture"
+        "Architecture",
+        "AddKeyValueAnnotationsAsText"
       ]
     },
     "option_type": {
diff --git a/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/MapValueToKey.cs b/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/MapValueToKey.cs
index 36dbd5de7f..a48827478f 100644
--- a/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/MapValueToKey.cs
+++ b/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/MapValueToKey.cs
@@ -8,7 +8,7 @@ internal partial class MapValueToKey
     {
         public override IEstimator<ITransformer> BuildFromOption(MLContext context, MapValueToKeyOption param)
         {
-            return context.Transforms.Conversion.MapValueToKey(param.OutputColumnName, param.InputColumnName);
+            return context.Transforms.Conversion.MapValueToKey(param.OutputColumnName, param.InputColumnName, addKeyValueAnnotationsAsText: param.AddKeyValueAnnotationsAsText);
         }
     }