Update notebook to show the error message

OpenMined · Jul 31, 2024 · 11fe274 · 11fe274
1 parent 98f650c
commit 11fe274
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/notebooks/api/0.8/13-forgot-user-password.ipynb b/notebooks/api/0.8/13-forgot-user-password.ipynb
@@ -89,7 +89,8 @@
     "guest_client = server.login_as_guest()\n",
     "res = guest_client.users.forgot_password(email=\"[email protected]\")\n",
     "\n",
-    "assert isinstance(res, SyftSuccess)"
+    "if not isinstance(res, SyftSuccess):\n",
+    "    raise Exception(f\"Res isn't SyftSuccess,  its {res}\")"
    ]
   },
   {
@@ -109,7 +110,10 @@
    "source": [
     "temp_token = datasite_client.users.request_password_reset(\n",
     "    datasite_client.notifications[0].linked_obj.resolve.id\n",
-    ")"
+    ")\n",
+    "\n",
+    "if not isinstance(temp_token, str):\n",
+    "    raise Exception(f\"temp_token isn't a string,  its {res}\")"
    ]
   },
   {