Update / Fix code generator notebook

TCord · Aug 19, 2022 · 248e8c9 · 248e8c9
1 parent d295dd4
commit 248e8c9
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/notebooks/mixture_generator_code_generator.ipynb b/notebooks/mixture_generator_code_generator.ipynb
@@ -181,6 +181,7 @@
     "    \n",
     "    # Plot and display\n",
     "    display_code = '\\n# Plot meeting\\n'\n",
+    "    display_code += 'from mms_msg.visualization.plot import plot_mixture\\n'\n",
     "    display_code += f'plot_mixture(ds[{example}])\\n\\n'\n",
     "\n",
     "    # Load & play\n",
@@ -197,7 +198,7 @@
     "    example['audio_data'] = pb.io.audioread.recursive_load_audio(example['audio_path'])\n",
     "    return example\n",
     "ds = ds.map(load_audio)\n",
-    "ds = ds.map(functools.partial({scenario_map_fn}, snr_range={snr_range}))\n",
+    "ds = ds.map(functools.partial({scenario_map_fn}))\n",
     "ds = ds.map(mms_msg.simulation.noise.white_microphone_noise)\n",
     "example = ds[{example}]\n",
     "\n",
@@ -501,9 +502,9 @@
    "hash": "916dbcbb3f70747c44a77c7bcd40155683ae19c65e1c03b4aa3499c5328201f1"
   },
   "kernelspec": {
-   "display_name": "Python (padertorch)",
+   "display_name": "base",
    "language": "python",
-   "name": "padertorch"
+   "name": "base"
   },
   "language_info": {
    "codemirror_mode": {
@@ -515,7 +516,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.9.12"
   }
  },
  "nbformat": 4,