From ab799db14d0ba64d6b06700a5b21cf9dec618c2b Mon Sep 17 00:00:00 2001
From: Safouane Chergui <31566323+chsafouane@users.noreply.github.com>
Date: Thu, 24 Feb 2022 11:27:27 +0100
Subject: [PATCH] Correcting merge of model perf & aequitas results

`model_uuid` is a column in both dataframes. It is not an index in aequitas results dataframe.
---
 notebooks/bias_reduction.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/notebooks/bias_reduction.ipynb b/notebooks/bias_reduction.ipynb
index fa35d2b..2e71d6d 100644
--- a/notebooks/bias_reduction.ipynb
+++ b/notebooks/bias_reduction.ipynb
@@ -159,7 +159,7 @@
     "    scatter_schema = ['model_uuid', performance_col, 'attribute_name', 'attribute_value', bias_metric, disparity_metric, 'model_tag']\n",
     "    if flip_disparity:\n",
     "        disparity_df[disparity_metric]= disparity_df.apply(lambda x: 1/x[disparity_metric] , axis=1)\n",
-    "    scatter = pd.merge(evals_df, disparity_df, how='left', on=['model_uuid'], left_index=True, sort=True, copy=True)\n",
+    "    scatter = pd.merge(evals_df, disparity_df, how='left', on=['model_uuid'], sort=True, copy=True)\n",
     "    scatter = scatter[['model_uuid', performance_col, 'attribute_name', 'attribute_value', bias_metric, disparity_metric]].copy()\n",
     "    scatter['model_tag'] = 'Other Models'\n",
     "    scatter.sort_values('model_precision', ascending = False, inplace=True, ignore_index=True)\n",