From 434f1fb86c928b27581bed8ad64b790975ad1e74 Mon Sep 17 00:00:00 2001 From: Bovard Doerschuk-Tiberi Date: Thu, 23 May 2024 08:46:08 -1000 Subject: [PATCH] CR --- .../envs/llm_20_questions/llm_20_questions.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) 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 7504ab4d..ed9b8234 100644 --- a/kaggle_environments/envs/llm_20_questions/llm_20_questions.py +++ b/kaggle_environments/envs/llm_20_questions/llm_20_questions.py @@ -200,12 +200,11 @@ def interpreter(state, env): else: increment_turn(active2, inactive2, step, guessed) - # make sure to end the game if either team guessed - if one_guessed or two_guessed: - if one_guessed and not two_guessed: - end_game(active2, inactive2, 0, DONE, DONE) - elif two_guessed and not one_guessed: - end_game(active1, inactive1, 0, DONE, DONE) + # make sure to end the game if only one team guessed correctly this round + if one_guessed and not two_guessed: + end_game(active2, inactive2, 0, DONE, DONE) + elif two_guessed and not one_guessed: + end_game(active1, inactive1, 0, DONE, DONE) return state