In WIT, set_target sets ground truth feature (#2314)

set_target now sets the ground truth feature in WIT as well. Before, user had to select ground truth feature in Fairness&Performance panel even after setting it in WitConfig.

Author: tolga-b

tensorboard/plugins/interactive_inference/witwidget/notebook/colab/wit.py

@@ -148,6 +148,9 @@ def infer_mutants(wit_id, details):
           wit.multiClass = config['multiclass'];
         }}
         wit.updateNumberOfModels();
+        if ('target_feature' in config) {{
+          wit.selectedLabelFeature = config['target_feature'];
+        }}
       }};
       window.updateExamplesCallback = examples => {{
         if (!wit.updateExampleContents) {{

tensorboard/plugins/interactive_inference/witwidget/notebook/jupyter/js/lib/wit.js

@@ -212,6 +212,9 @@ var WITView = widgets.DOMWidgetView.extend({
       this.view_.multiClass = config['multiclass'];
     }
     this.view_.updateNumberOfModels();
+    if ('target_feature' in config) {
+      this.view_.selectedLabelFeature = config['target_feature'];
+    }
   },
   spriteChanged: function() {
     if (!this.setupComplete) {