Update WitWidget pip version (#2283)

tensorflow · May 28, 2019 · 39c80de · 39c80de
1 parent 399b86b
commit 39c80de
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tensorboard/plugins/interactive_inference/utils/inference_utils.py b/tensorboard/plugins/interactive_inference/utils/inference_utils.py
@@ -768,8 +768,8 @@ def run_inference(examples, serving_bundle):
     attributions = None
     # If the custom prediction function returned a dict, then parse out the
     # prediction scores and the attributions. If it is just a list, then the
-    # results are just the prediction results without attributions.
-    if not isinstance(values, list):
+    # results are the prediction results without attributions.
+    if isinstance(values, dict):
       attributions = values['attributions']
       values = values['predictions']
     return (common_utils.convert_prediction_values(values, serving_bundle),

diff --git a/tensorboard/plugins/interactive_inference/witwidget/version.py b/tensorboard/plugins/interactive_inference/witwidget/version.py
@@ -14,4 +14,4 @@
 
 """Contains the version string."""
 
-VERSION = '1.0.2'
+VERSION = '1.1'