fixing broken import in test

TLSDC · TLSDC · commit 544908e890fa · 2025-05-02T16:30:57.000-04:00
diff --git a/tests/agents/test_generic_prompt.py b/tests/agents/test_generic_prompt.py
@@ -2,7 +2,7 @@
 
 import bgym
 import pytest
-from bgym import HighLevelActionSetArgs
+from bgym import HighLevelActionSet, HighLevelActionSetArgs
 
 from agentlab.agents import dynamic_prompting as dp
 from agentlab.agents.generic_agent.agent_configs import FLAGS_GPT_3_5
