Merge pull request #2205 from felipetovarhenao:patch-2

PiperOrigin-RevId: 518323478

Author: copybara-github

site/en/tutorials/audio/music_generation.ipynb

@@ -145,7 +145,7 @@
         "\n",
         "from IPython import display\n",
         "from matplotlib import pyplot as plt\n",
-        "from typing import Dict, List, Optional, Sequence, Tuple"
+        "from typing import Optional"
       ]
     },
     {
@@ -1073,8 +1073,8 @@
         "def predict_next_note(\n",
         "    notes: np.ndarray, \n",
         "    keras_model: tf.keras.Model, \n",
-        "    temperature: float = 1.0) -> int:\n",
-        "  \"\"\"Generates a note IDs using a trained sequence model.\"\"\"\n",
+        "    temperature: float = 1.0) -> tuple[int, float, float]:\n",
+        "  \"\"\"Generates a note as a tuple of (pitch, step, duration), using a trained sequence model.\"\"\"\n",
         "\n",
         "  assert temperature > 0\n",
         "\n",