Add missing dependencies in Agile Modeling notebook.

PiperOrigin-RevId: 617513620

agile_modeling.ipynb

@@ -30,8 +30,10 @@
    "source": [
     " #@title Imports. { vertical-output: true }\n",
     "\n",
+    "import collections\n",
     "from etils import epath\n",
     "from ml_collections import config_dict\n",
+    "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import tensorflow as tf\n",
     "import tqdm\n",
@@ -40,7 +42,14 @@
     "\n",
     "from chirp import audio_utils\n",
     "from chirp.inference import embed_lib\n",
-    "from chirp.inference import tf_examples\n"
+    "from chirp.inference import tf_examples\n",
+    "from chirp.inference import models\n",
+    "from chirp.models import metrics\n",
+    "from chirp.inference.search import bootstrap\n",
+    "from chirp.inference.search import search\n",
+    "from chirp.inference.search import display\n",
+    "from chirp.inference.classify import classify\n",
+    "from chirp.inference.classify import data_lib\n"
    ]
   },
   {
@@ -91,7 +100,7 @@
     "  if (bootstrap_config.model_key == 'separate_embed_model'\n",
     "      and not separation_model_path.strip()):\n",
     "    separation_model_key = 'separator_model_tf'\n",
-    "    separation_model_path = config.model_config.separator_model_tf_config.model_path\n",
+    "    separation_model_path = bootstrap_config.model_config.separator_model_tf_config.model_path\n",
     "\n",
     "# Load separation model.\n",
     "if separation_model_path:\n",
@@ -265,7 +274,7 @@
     "  break\n",
     "\n",
     "# Load/refresh bootstrap_config for subsequent steps.\n",
-    "print(f'\\nRefreshing bootstrap_config.', flush=True)\n",
+    "print('\\nRefreshing bootstrap_config.', flush=True)\n",
     "bootstrap_config = bootstrap.BootstrapConfig.load_from_embedding_config(\n",
     "    embeddings_path=embeddings_path,\n",
     "    annotated_path=labeled_data_path)\n",