From 3ef5755553a6c67e0b73c75429e46fe63fd0bead Mon Sep 17 00:00:00 2001
From: Ifeanyi Obinelo <62487964+kz4killua@users.noreply.github.com>
Date: Tue, 6 Jun 2023 15:33:27 +0100
Subject: [PATCH] Update notebook to work with new version of Capsa

I fixed one cell which no longer worked due to the new Capsa version.

Thanks.
---
 lab3/Part1_IntroductionCapsa.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lab3/Part1_IntroductionCapsa.ipynb b/lab3/Part1_IntroductionCapsa.ipynb
index e6871272..89494c8b 100644
--- a/lab3/Part1_IntroductionCapsa.ipynb
+++ b/lab3/Part1_IntroductionCapsa.ipynb
@@ -350,11 +350,11 @@
         "### Generate and visualize bias scores for data in test set ###\n",
         "\n",
         "# Call the risk-aware model to generate scores\n",
-        "predictions, bias = bias_wrapped_dense_NN(x_test)\n",
+        "predictions = bias_wrapped_dense_NN(x_test)\n",
         "\n",
         "# Visualize the relationship between the input data x and the bias\n",
         "fig, ax = plt.subplots(2, 1, figsize=(8,6))\n",
-        "ax[0].plot(x_test, bias, label='bias')\n",
+        "ax[0].plot(x_test, predictions.bias, label='bias')\n",
         "ax[0].set_ylabel('Estimated Bias')\n",
         "ax[0].legend()\n",
         "\n",