From 8a953f3cc69ee154207c74290878f3165761f3d0 Mon Sep 17 00:00:00 2001
From: Yufeng Zhang <30943549+yufengzh@users.noreply.github.com>
Date: Sun, 23 Jul 2023 12:11:11 -0400
Subject: [PATCH] Fix to unpack return value in Lab 2 Part 2

The `ss_vae_loss_function` returns a length-3 tuple, namely `loss`, `class_loss`, and `vae_loss`. Therefore, we add a third one when calling this function.
---
 lab2/Part2_FaceDetection.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lab2/Part2_FaceDetection.ipynb b/lab2/Part2_FaceDetection.ipynb
index c877149c..59678b73 100644
--- a/lab2/Part2_FaceDetection.ipynb
+++ b/lab2/Part2_FaceDetection.ipynb
@@ -779,7 +779,7 @@
         "    y_logit, z_mean, z_logsigma, x_recon = ss_vae(x)\n",
         "\n",
         "    '''TODO: call the SS_VAE loss function to compute the loss'''\n",
-        "    loss, class_loss = ss_vae_loss_function('''TODO arguments''') # TODO\n",
+        "    loss, class_loss, _ = ss_vae_loss_function('''TODO arguments''') # TODO\n",
         "  \n",
         "  '''TODO: use the GradientTape.gradient method to compute the gradients.\n",
         "     Hint: this is with respect to the trainable_variables of the SS_VAE.'''\n",