Fixing bug in log parsing

Author: Fiona-Waters

tests/trainer/resources/mnist.ipynb

@@ -513,12 +513,10 @@
     "job = client.get_job(name=job_name)\n",
     "pod_logs = client.get_job_logs(name=job_name, follow=False)\n",
     "\n",
-    "# Flatten all pod logs into a single list of lines\n",
-    "logs = []\n",
-    "for log_line in pod_logs:\n",
-    "    logs.extend(str(log_line).splitlines())\n",
+    "# Collect all log lines from the generator into a list\n",
+    "logs = list(pod_logs)\n",
+    "log_text = \"\\n\".join(str(line) for line in logs)\n",
     "\n",
-    "log_text = \"\\n\".join(logs)\n",
     "\n",
     "print(f\"Training job final status: {job.status}\")\n",
     "\n",