Fix missing import

microsoft · Feb 12, 2025 · 5ff7e1b · 5ff7e1b
1 parent 1f4c095
commit 5ff7e1b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/unit_tests/test_llm.py b/tests/unit_tests/test_llm.py
@@ -1,5 +1,6 @@
 from opto.utils.llm import LLM
 from opto.optimizers.utils import print_color
+import os
 
 if os.path.exists("OAI_CONFIG_LIST") or os.environ.get("DEFAULT_LITELLM_MODEL") or os.environ.get("OPENAI_API_KEY"):
     llm = LLM()