Fix anchor tabular explanation serialization (kubeflow#443)

* Fix anchor tabular explanation serialization * Update fix for anchor tabular
magdalenakuhn17 · Oct 18, 2019 · a376b5d · a376b5d
1 parent 8fa0ca6
commit a376b5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/alibiexplainer/alibiexplainer/anchor_tabular.py b/python/alibiexplainer/alibiexplainer/anchor_tabular.py
@@ -96,7 +96,7 @@ def explain(self, inputs: List) -> Dict:
                         anchor_exp['raw']['examples'][i][key] = parr.values
 
                 instance = anchor_exp['raw']['instance']
-                anchor_exp['raw']['instance'] = pd.DataFrame(instance).replace(self.cmap).values
+                anchor_exp['raw']['instance'] = pd.DataFrame([instance]).replace(self.cmap).values.squeeze()
             return anchor_exp
         else:
             raise Exception("Explainer not initialized")