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