From 9b43d93e9c5cc2659356860f69d5945f91d17591 Mon Sep 17 00:00:00 2001 From: Bovard Doerschuk-Tiberi Date: Fri, 24 May 2024 06:27:23 -1000 Subject: [PATCH] make sure the answerer agents get they keyword and category --- kaggle_environments/envs/llm_20_questions/llm_20_questions.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/kaggle_environments/envs/llm_20_questions/llm_20_questions.py b/kaggle_environments/envs/llm_20_questions/llm_20_questions.py index 45121dd8..d08d9883 100644 --- a/kaggle_environments/envs/llm_20_questions/llm_20_questions.py +++ b/kaggle_environments/envs/llm_20_questions/llm_20_questions.py @@ -87,6 +87,8 @@ def answerer_agent(obs): agents = {GUESSER: guesser_agent, ANSWERER: answerer_agent} def guesser_action(active, inactive, step): + inactive.observation.keyword = keyword + inactive.observation.category = category guessed = False if not active.action: active.status = ERROR