From bea7dc2ca6e3cb4ae0bdb3da1ed903edf630fde5 Mon Sep 17 00:00:00 2001 From: Helen Date: Thu, 30 Apr 2026 00:11:27 -0400 Subject: [PATCH 1/4] added ogbench as a submodule --- .gitmodules | 3 +++ ogbench | 1 + 2 files changed, 4 insertions(+) create mode 100644 .gitmodules create mode 160000 ogbench diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..0f82088 --- /dev/null +++ b/.gitmodules @@ -0,0 +1,3 @@ +[submodule "ogbench"] + path = ogbench + url = ./ogbench diff --git a/ogbench b/ogbench new file mode 160000 index 0000000..d750941 --- /dev/null +++ b/ogbench @@ -0,0 +1 @@ +Subproject commit d7509418e7a2243ebea3c5fc081ed57557e89719 From 145bf4541188b840a7efd51f2007b092d606e5b7 Mon Sep 17 00:00:00 2001 From: Helen Date: Thu, 30 Apr 2026 00:57:05 -0400 Subject: [PATCH 2/4] fixed some D2 mazes --- .vscode/launch.json | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 .vscode/launch.json diff --git a/.vscode/launch.json b/.vscode/launch.json new file mode 100644 index 0000000..690053b --- /dev/null +++ b/.vscode/launch.json @@ -0,0 +1,21 @@ +{ + // Use IntelliSense to learn about possible attributes. + // Hover to view descriptions of existing attributes. + // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387 + "version": "0.2.0", + "configurations": [ + { + "name": "Python Debugger: Current File", + "type": "debugpy", + "request": "launch", + "program": "${file}", + "console": "integratedTerminal", + // add python paths + "env": { + "PYTHONPATH": "${workspaceFolder}/ogbench:${workspaceFolder}/ogbench/ogbench:${workspaceFolder}/ogbench/ogbench/procgen" + } + + }, + + ] +} \ No newline at end of file From 577338d08be2ef40d96014294a52bf0f948b70fb Mon Sep 17 00:00:00 2001 From: Helen Date: Fri, 15 May 2026 10:33:33 -0400 Subject: [PATCH 3/4] exp 3 prompts --- prompts/__init__.py | 1 + prompts/__pycache__/__init__.cpython-312.pyc | Bin 0 -> 218 bytes .../condition_set_1_prompt.cpython-312.pyc | Bin 0 -> 1737 bytes ..._set_6_in_context_learning.cpython-312.pyc | Bin 0 -> 1519 bytes prompts/condition_set_1_prompt.py | 62 ++++++++++++++ prompts/condition_set_2_observation_format.py | 56 +++++++++++++ prompts/condition_set_3_context_window.py | 68 ++++++++++++++++ prompts/condition_set_4_action_space.py | 61 ++++++++++++++ prompts/condition_set_5_querying_strategy.py | 59 ++++++++++++++ .../condition_set_6_in_context_learning.py | 76 ++++++++++++++++++ 10 files changed, 383 insertions(+) create mode 100644 prompts/__init__.py create mode 100644 prompts/__pycache__/__init__.cpython-312.pyc create mode 100644 prompts/__pycache__/condition_set_1_prompt.cpython-312.pyc create mode 100644 prompts/__pycache__/condition_set_6_in_context_learning.cpython-312.pyc create mode 100644 prompts/condition_set_1_prompt.py create mode 100644 prompts/condition_set_2_observation_format.py create mode 100644 prompts/condition_set_3_context_window.py create mode 100644 prompts/condition_set_4_action_space.py create mode 100644 prompts/condition_set_5_querying_strategy.py create mode 100644 prompts/condition_set_6_in_context_learning.py diff --git a/prompts/__init__.py b/prompts/__init__.py new file mode 100644 index 0000000..b96834c --- /dev/null +++ b/prompts/__init__.py @@ -0,0 +1 @@ +"""Prompt condition sets for maze evaluation experiments.""" diff --git a/prompts/__pycache__/__init__.cpython-312.pyc b/prompts/__pycache__/__init__.cpython-312.pyc new file mode 100644 index 0000000000000000000000000000000000000000..f3a8235227cd83d13e9b8a0d7ff5c6ccb8ac55e7 GIT binary patch literal 218 zcmX@j%ge<81XX(MS++p>F^B^Lj8MjB4j^MXLkdF_LnWgoQ)?JCxfkI&4@EQycT mE2#X%VUwGmQks)$SHuQ11mw(OkP|;JGcq#XVz4P<0dfF2uRN~+ literal 0 HcmV?d00001 diff --git a/prompts/__pycache__/condition_set_1_prompt.cpython-312.pyc b/prompts/__pycache__/condition_set_1_prompt.cpython-312.pyc new file mode 100644 index 0000000000000000000000000000000000000000..886a234d5315dcb0cec7490fc66b156f12f0c964 GIT binary patch literal 1737 zcmd5-Pj4GV6ko67*xix@!wIBHPeoJ&=$fPiM5rn`woz9kYb!RP^<_2Lnb-rfJ7#9q zsf{YZkx#&-Ux3fT7YKW*#Fc}q-g@HA+AdPnGvLE|J@el0{rkP2f8W~j9Q@4v>UXf| zIDgt7>rZv{^87=5`PHG$(4nPaiMm6Vmgxqq(9MxctE|LaT6^NsYfmb}^32=4KEK(L zDdmRC6f`ricL)E5Lqf8J5WW-VfT8b%8 zM?fGUbLIm&C`h>07Mg1?5>hhdBVrIl7C0lt7>4~JLtx+Y5*Ck1%5|bK2c{A9(EIVE zIqXFJUZ)j&6||z0PV0XEvFA0XgbQ*m7?9ZJuFUU1hn%40#yaznN+TcPe*lz<#Dcos)3E^>Z5*~YAkWLX(su@BB z;aMjfbWb1JU}GsY1NI2B6EoY(B#3mO#Ag@+}jp$}(l3}VT`r3XkYsB^? z*e3Fq`+u88n;9Gjl0s#I;VAZe_?pf1vH&^38B-~hhD&b7$PN>aEm2}A@X+`6SJswY zF^)1z5Q!)Fqdr747Aly+Weq2YrJh!bThxU4W0^ zrGx-wI#zsAhyz}%R!DGd<~Pp@X=8`5_D2KuXd-yb&630O{on#2gduPj4&MJ5GF<1i zHO}Y)?!7!-q4~RRUPGD`3FWml-fks#In6ih^5)lLlvbj+Mw;g}%3^NSF|RHqSBJYB zc{PftjH4*u=?|J=yLsAAwF-V->CON?W7lxj&h`sNGE@#a+Iq-a<%Je@8d* zu_%{Hr9Yex78~`qH-6e#+ADkZ>cvCSK}i4r literal 0 HcmV?d00001 diff --git a/prompts/__pycache__/condition_set_6_in_context_learning.cpython-312.pyc b/prompts/__pycache__/condition_set_6_in_context_learning.cpython-312.pyc new file mode 100644 index 0000000000000000000000000000000000000000..b0de01938ecc41a5e5aeec4f1cbec63a22166229 GIT binary patch literal 1519 zcmb_c&2HO95ay3 zX(l*QRHKJ`i1FQ2a7|xmlu;}>}GFQ@4YbEa~I z3P}~^de8eI^q(Yy{y0d6hk+k{8~Vv0@DGMRdS1W6EW^iuwL(x}y&n(yVf4YL4;CS>Agqw{b>uL58bqV`=viWJ`|C^_*EO^` zzZ&4RrStd@QV@;Fb z`k5#Zma9ui3DS~bJ}omW5uO5HjjyrCD-Q$?66H)OGZ9Q7EfB{Qn_>+hq6spUl)|)p zUji_MxU~2GSKs|oN|UDeSN|{0%0Fh|BC}^x0O$Tp2t^Hdsm|soj)L|0HHG05pihdt zDG^^BP^dF5bx}el9DEJefMPh}I0qK$ZP3h0M+xXCdAPV0{#Qqf55`oAT{RPW4{>qU zMzz?#zdjVaNwCqGmiReM%}Ap=giR*2VZx~I9#Rk$oimU0yr4Uam*E7hfDn;52eqU) zgKSw|NKsMHCY2K*bIenKI2KxGl+*O2=`&U*sN{r7r7={(r}rR3jZ$3TW0o;J@1lJQ zn$H;rq{Uess8VP`Pn-0+i@Tut^Qur@D&2sThV}|)Wh3Bb(p0x@L$QhBVrQ#%7^yc5 zmG$*h-lnPw5yyVZm4XT3Fe??OWp z#*_74OwRXWF@M{BlnYX3^ecG-@NM|1ui 5%, use Verbose.", + "prompts": { + "standard": STANDARD_PROMPT, + "explicit": EXPLICIT_PROMPT, + }, +} diff --git a/prompts/condition_set_2_observation_format.py b/prompts/condition_set_2_observation_format.py new file mode 100644 index 0000000..77cd325 --- /dev/null +++ b/prompts/condition_set_2_observation_format.py @@ -0,0 +1,56 @@ +"""Condition set 2: observation format.""" + +IMAGE_PLUS_TEXT_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Available actions: +{DOMAIN_SPECIFIC_ACTION_LIST} + +Current observation: +{DOMAIN_SPECIFIC_OBSERVATION} + +Observation description: +{OBSERVATION_TEXT_DESCRIPTION} + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +""" + +IMAGE_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Available actions: +{DOMAIN_SPECIFIC_ACTION_LIST} + +Current observation: +{DOMAIN_SPECIFIC_OBSERVATION} + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +""" + +CONDITION_SET = { + "id": "condition_set_2", + "name": "Observation format", + "comparisons": [ + "Image + text prompt", + "Image only (no text)", + ], + "decision": "Does text add meaningful signal?", + "prompts": { + "image_plus_text": IMAGE_PLUS_TEXT_PROMPT, + "image": IMAGE_PROMPT, + }, +} diff --git a/prompts/condition_set_3_context_window.py b/prompts/condition_set_3_context_window.py new file mode 100644 index 0000000..cfdc2c3 --- /dev/null +++ b/prompts/condition_set_3_context_window.py @@ -0,0 +1,68 @@ +"""Condition set 3: context window.""" + +HISTORY_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Available actions: +{DOMAIN_SPECIFIC_ACTION_LIST} + +Observation history: + +Frame t-2: +{FRAME_T_MINUS_2} + +Frame t-1: +{FRAME_T_MINUS_1} + +Current frame: +{CURRENT_FRAME} + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +""" + +TEXT_SUMMARY_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Available actions: +{DOMAIN_SPECIFIC_ACTION_LIST} + +Current observation: +{CURRENT_FRAME} + +Exploration summary: +{MECHANISMS_INTERACTED_WITH}.{SUBGOALS_ACHIEVED}.{PATH_IN_LAST_10_FRAMES}. +# Example: you've interacted with the yellow key and the yellow door. +# You've opened the yellow door. +# In the last 10 frames, you've traveled from [1, 1] to [1, 10] + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +""" + +CONDITION_SET = { + "id": "condition_set_3", + "name": "Context window", + "comparisons": [ + "0 history (current frame only)", + "Last 3 frames", + "Current frame + text summary of prior actions", + ], + "decision": "Is there a cheap alternative to feeding multiple frames?", + "prompts": { + "history": HISTORY_PROMPT, + "text_summary": TEXT_SUMMARY_PROMPT, + }, +} diff --git a/prompts/condition_set_4_action_space.py b/prompts/condition_set_4_action_space.py new file mode 100644 index 0000000..a33670e --- /dev/null +++ b/prompts/condition_set_4_action_space.py @@ -0,0 +1,61 @@ +"""Condition set 4: action space.""" + +EGOCENTRIC_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Available actions: +- TURN_LEFT +- TURN_RIGHT +- MOVE_FORWARD +- INTERACT + +Current observation: +{DOMAIN_SPECIFIC_OBSERVATION} + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +Output only the action name. +""" + +CARDINAL_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Available actions: +- MOVE_NORTH +- MOVE_SOUTH +- MOVE_EAST +- MOVE_WEST +- INTERACT + +Current observation: +{DOMAIN_SPECIFIC_OBSERVATION} + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +""" + +CONDITION_SET = { + "id": "condition_set_4", + "name": "Action space", + "comparisons": [ + "Egocentric: TURN_LEFT, TURN_RIGHT, MOVE_FORWARD, INTERACT", + "Cardinal: MOVE_NORTH/SOUTH/EAST/WEST, INTERACT", + ], + "decision": "If delta is trivial (<5%), go egocentric. If massive (>15%), reassess.", + "prompts": { + "egocentric": EGOCENTRIC_PROMPT, + "cardinal": CARDINAL_PROMPT, + }, +} diff --git a/prompts/condition_set_5_querying_strategy.py b/prompts/condition_set_5_querying_strategy.py new file mode 100644 index 0000000..1466e94 --- /dev/null +++ b/prompts/condition_set_5_querying_strategy.py @@ -0,0 +1,59 @@ +"""Condition set 5: querying strategy.""" + +SUBGOAL_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Current observation: +{DOMAIN_SPECIFIC_OBSERVATION} + +Inventory: +{INVENTORY} + +Before acting, produce a short high-level plan. + +Your plan should identify: +1. important keys +2. important doors or gates +3. likely exploration order +4. important switches or bottlenecks + +Output concise numbered subgoals only. +""" + +SUBGOAL_EXECUTION_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Current high-level plan: +{SUBGOAL_PLAN} + +Available actions: +{DOMAIN_SPECIFIC_ACTION_LIST} + +Current observation: +{DOMAIN_SPECIFIC_OBSERVATION} + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +""" + +CONDITION_SET = { + "id": "condition_set_5", + "name": "Querying strategy", + "comparisons": [ + "Step-by-step: one action per query", + "Subgoal planning: model outputs plan first, then executes per-subgoal", + ], + "decision": "Does planning help? If yes, benchmark tests planning or execution?", + "prompts": { + "subgoal": SUBGOAL_PROMPT, + "subgoal_execution": SUBGOAL_EXECUTION_PROMPT, + }, +} diff --git a/prompts/condition_set_6_in_context_learning.py b/prompts/condition_set_6_in_context_learning.py new file mode 100644 index 0000000..9009d67 --- /dev/null +++ b/prompts/condition_set_6_in_context_learning.py @@ -0,0 +1,76 @@ +"""Condition set 6: in-context learning.""" + +ONE_SHOT_PROMPT = """Example maze interaction: + +mechanisms present: +{ICL_MAZE_SPECIFIC_MECHS} + +Available actions: +{DOMAIN_SPECIFIC_ACTION_LIST} + +Observation: +{ICL_OBSERVATION_1} + +Inventory: +{ICL_INVENTORY_1} + +Action: +{ICL_ACTION_1} + +Observation: +{ICL_OBSERVATION_2} + +Inventory: +{ICL_INVENTORY_2} + +Action: +{ICL_ACTION_2} + +Observation: +{ICL_OBSERVATION_3} + +Inventory: +{ICL_INVENTORY_3} + +Action: +{ICL_ACTION_3} + +End of example. + +Now solve the following maze. + +You are the red triangular agent solving a maze. Your mission is to navigate to the green square. + +mechanisms present: +{MAZE_SPECIFIC_MECHS} + +Available actions: +{DOMAIN_SPECIFIC_ACTION_LIST} + +Current observation: +{DOMAIN_SPECIFIC_OBSERVATION} + +Inventory: +{INVENTORY} + +Choose exactly ONE action to take from the available actions. + +Output only the action name. +""" + +CONDITION_SET = { + "id": "condition_set_6", + "name": "In-context learning", + "comparisons": [ + "Zero-shot: no examples", + "1-shot: one example trajectory (different maze, same mechanism type)", + ], + "decision": ( + "If 1-shot dramatically improves performance, bottleneck is task " + "understanding, not capability. Determines zero-shot vs few-shot." + ), + "constraint": "ICL examples must not use evaluation mazes.", + "prompts": { + "one_shot": ONE_SHOT_PROMPT, + }, +} From 8c91c18a9592089695c005fa97f60fe12fbc37a3 Mon Sep 17 00:00:00 2001 From: Helen Date: Fri, 15 May 2026 12:30:58 -0400 Subject: [PATCH 4/4] moved the CONDITION_SET dict into exp_design.py --- prompts/__init__.py | 4 +++ prompts/condition_set_1_prompt.py | 3 +- prompts/condition_set_2_observation_format.py | 29 ++++--------------- prompts/condition_set_3_context_window.py | 3 +- prompts/condition_set_4_action_space.py | 3 +- prompts/condition_set_5_querying_strategy.py | 3 +- .../condition_set_6_in_context_learning.py | 6 +++- prompts/exp_design.py | 17 +++++++++++ 8 files changed, 40 insertions(+), 28 deletions(-) create mode 100644 prompts/exp_design.py diff --git a/prompts/__init__.py b/prompts/__init__.py index b96834c..80835a2 100644 --- a/prompts/__init__.py +++ b/prompts/__init__.py @@ -1 +1,5 @@ """Prompt condition sets for maze evaluation experiments.""" + +from .exp_design import CONDITION_SETS + +__all__ = ["CONDITION_SETS"] diff --git a/prompts/condition_set_1_prompt.py b/prompts/condition_set_1_prompt.py index 4565383..947e3b6 100644 --- a/prompts/condition_set_1_prompt.py +++ b/prompts/condition_set_1_prompt.py @@ -48,7 +48,6 @@ """ CONDITION_SET = { - "id": "condition_set_1", "name": "Prompt", "comparisons": [ "Standard: goal + mechanism descriptions + action list", @@ -60,3 +59,5 @@ "explicit": EXPLICIT_PROMPT, }, } + +PROMPTS = CONDITION_SET["prompts"] diff --git a/prompts/condition_set_2_observation_format.py b/prompts/condition_set_2_observation_format.py index 77cd325..070aaad 100644 --- a/prompts/condition_set_2_observation_format.py +++ b/prompts/condition_set_2_observation_format.py @@ -1,28 +1,10 @@ """Condition set 2: observation format.""" -IMAGE_PLUS_TEXT_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. +from .condition_set_1_prompt import STANDARD_PROMPT -mechanisms present: -{MAZE_SPECIFIC_MECHS} - -Available actions: -{DOMAIN_SPECIFIC_ACTION_LIST} - -Current observation: -{DOMAIN_SPECIFIC_OBSERVATION} - -Observation description: -{OBSERVATION_TEXT_DESCRIPTION} +IMAGE_PLUS_TEXT_PROMPT = STANDARD_PROMPT -Inventory: -{INVENTORY} - -Choose exactly ONE action to take from the available actions. - -Output only the action name. -""" - -IMAGE_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. +IMAGE_ONLY_PROMPT = """You are the red triangular agent solving a maze. Your mission is to navigate to the green square. mechanisms present: {MAZE_SPECIFIC_MECHS} @@ -42,7 +24,6 @@ """ CONDITION_SET = { - "id": "condition_set_2", "name": "Observation format", "comparisons": [ "Image + text prompt", @@ -51,6 +32,8 @@ "decision": "Does text add meaningful signal?", "prompts": { "image_plus_text": IMAGE_PLUS_TEXT_PROMPT, - "image": IMAGE_PROMPT, + "image_only": IMAGE_ONLY_PROMPT, }, } + +PROMPTS = CONDITION_SET["prompts"] diff --git a/prompts/condition_set_3_context_window.py b/prompts/condition_set_3_context_window.py index cfdc2c3..3d45331 100644 --- a/prompts/condition_set_3_context_window.py +++ b/prompts/condition_set_3_context_window.py @@ -53,7 +53,6 @@ """ CONDITION_SET = { - "id": "condition_set_3", "name": "Context window", "comparisons": [ "0 history (current frame only)", @@ -66,3 +65,5 @@ "text_summary": TEXT_SUMMARY_PROMPT, }, } + +PROMPTS = CONDITION_SET["prompts"] diff --git a/prompts/condition_set_4_action_space.py b/prompts/condition_set_4_action_space.py index a33670e..9ea8632 100644 --- a/prompts/condition_set_4_action_space.py +++ b/prompts/condition_set_4_action_space.py @@ -47,7 +47,6 @@ """ CONDITION_SET = { - "id": "condition_set_4", "name": "Action space", "comparisons": [ "Egocentric: TURN_LEFT, TURN_RIGHT, MOVE_FORWARD, INTERACT", @@ -59,3 +58,5 @@ "cardinal": CARDINAL_PROMPT, }, } + +PROMPTS = CONDITION_SET["prompts"] diff --git a/prompts/condition_set_5_querying_strategy.py b/prompts/condition_set_5_querying_strategy.py index 1466e94..cdcf7bf 100644 --- a/prompts/condition_set_5_querying_strategy.py +++ b/prompts/condition_set_5_querying_strategy.py @@ -45,7 +45,6 @@ """ CONDITION_SET = { - "id": "condition_set_5", "name": "Querying strategy", "comparisons": [ "Step-by-step: one action per query", @@ -57,3 +56,5 @@ "subgoal_execution": SUBGOAL_EXECUTION_PROMPT, }, } + +PROMPTS = CONDITION_SET["prompts"] diff --git a/prompts/condition_set_6_in_context_learning.py b/prompts/condition_set_6_in_context_learning.py index 9009d67..962ea21 100644 --- a/prompts/condition_set_6_in_context_learning.py +++ b/prompts/condition_set_6_in_context_learning.py @@ -1,5 +1,7 @@ """Condition set 6: in-context learning.""" +from .condition_set_1_prompt import STANDARD_PROMPT + ONE_SHOT_PROMPT = """Example maze interaction: mechanisms present: @@ -59,7 +61,6 @@ """ CONDITION_SET = { - "id": "condition_set_6", "name": "In-context learning", "comparisons": [ "Zero-shot: no examples", @@ -71,6 +72,9 @@ ), "constraint": "ICL examples must not use evaluation mazes.", "prompts": { + "zero_shot": STANDARD_PROMPT, "one_shot": ONE_SHOT_PROMPT, }, } + +PROMPTS = CONDITION_SET["prompts"] diff --git a/prompts/exp_design.py b/prompts/exp_design.py new file mode 100644 index 0000000..3bd3260 --- /dev/null +++ b/prompts/exp_design.py @@ -0,0 +1,17 @@ +"""Experiment 3 prompt condition-set registry.""" + +from .condition_set_1_prompt import CONDITION_SET as CONDITION_SET_1 +from .condition_set_2_observation_format import CONDITION_SET as CONDITION_SET_2 +from .condition_set_3_context_window import CONDITION_SET as CONDITION_SET_3 +from .condition_set_4_action_space import CONDITION_SET as CONDITION_SET_4 +from .condition_set_5_querying_strategy import CONDITION_SET as CONDITION_SET_5 +from .condition_set_6_in_context_learning import CONDITION_SET as CONDITION_SET_6 + +CONDITION_SETS = { + CONDITION_SET_1["name"]: CONDITION_SET_1, + CONDITION_SET_2["name"]: CONDITION_SET_2, + CONDITION_SET_3["name"]: CONDITION_SET_3, + CONDITION_SET_4["name"]: CONDITION_SET_4, + CONDITION_SET_5["name"]: CONDITION_SET_5, + CONDITION_SET_6["name"]: CONDITION_SET_6, +}