Skip to content

Commit 87d1431

Browse files
authored
make sure the answerer agents get they keyword and category (#267)
1 parent ea3bea2 commit 87d1431

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

kaggle_environments/envs/llm_20_questions/llm_20_questions.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,8 @@ def answerer_agent(obs):
8787
agents = {GUESSER: guesser_agent, ANSWERER: answerer_agent}
8888

8989
def guesser_action(active, inactive, step):
90+
inactive.observation.keyword = keyword
91+
inactive.observation.category = category
9092
guessed = False
9193
if not active.action:
9294
active.status = ERROR

0 commit comments

Comments
 (0)