Merge pull request #2190 from lucifertrj:audio_change

PiperOrigin-RevId: 514691777

Author: copybara-github

site/en/tutorials/audio/simple_audio.ipynb

@@ -322,24 +322,17 @@
       },
       "outputs": [],
       "source": [
+        "plt.figure(figsize=(16, 10))\n",
         "rows = 3\n",
         "cols = 3\n",
         "n = rows * cols\n",
-        "fig, axes = plt.subplots(rows, cols, figsize=(16, 9))\n",
-        "\n",
         "for i in range(n):\n",
-        "  if i>=n:\n",
-        "    break\n",
-        "  r = i // cols\n",
-        "  c = i % cols\n",
-        "  ax = axes[r][c]\n",
-        "  ax.plot(example_audio[i].numpy())\n",
-        "  ax.set_yticks(np.arange(-1.2, 1.2, 0.2))\n",
-        "  label = label_names[example_labels[i]]\n",
-        "  ax.set_title(label)\n",
-        "  ax.set_ylim([-1.1,1.1])\n",
-        "\n",
-        "plt.show()"
+        "  plt.subplot(rows, cols, i+1)\n",
+        "  audio_signal = example_audio[i]\n",
+        "  plt.plot(audio_signal)\n",
+        "  plt.title(label_names[example_labels[i]])\n",
+        "  plt.yticks(np.arange(-1.2, 1.2, 0.2))\n",
+        "  plt.ylim([-1.1, 1.1])"
       ]
     },
     {