fix import error (#2797)

wandb · Oct 28, 2024 · 7e94360 · 7e94360
1 parent f79fbcc
commit 7e94360
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/docs/docs/reference/gen_notebooks/audio_with_weave.md b/docs/docs/reference/gen_notebooks/audio_with_weave.md
@@ -45,6 +45,7 @@ Next, load the required API keys for OpenAI and Weave. Here, we use set_env whic
 
 ```python
 # Set environment variables.
+from set_env import set_env
 
 _ = set_env("OPENAI_API_KEY")
 _ = set_env("WANDB_API_KEY")

diff --git a/docs/notebooks/audio_with_weave.ipynb b/docs/notebooks/audio_with_weave.ipynb
@@ -79,6 +79,7 @@
    "outputs": [],
    "source": [
     "# Set environment variables.\n",
+    "from set_env import set_env\n",
     "\n",
     "_ = set_env(\"OPENAI_API_KEY\")\n",
     "_ = set_env(\"WANDB_API_KEY\")"