kaggle-environments 1.14.11__py2.py3-none-any.whl → 1.14.12__py2.py3-none-any.whl

kaggle_environments/__init__.py

@@ -21,7 +21,7 @@ from .main import http_request
 from . import errors
 from . import utils
 
-__version__ = "1.14.11"
+__version__ = "1.14.12"
 
 __all__ = ["Agent", "environments", "errors", "evaluate", "http_request",
            "make", "register", "utils", "__version__",

kaggle_environments/envs/llm_20_questions/llm_20_questions.js

@@ -89,8 +89,10 @@ async function renderer(context) {
     }
 
     const info = environment.info;
-    const team1_text = info?.TeamNames?.[0] || "Team 1";
-    const team2_text = info?.TeamNames?.[2] || "Team 2";
+    const team1_text = info?.TeamNames?.[0] || "Team 1 Agent 1";
+    const team1_text_two = info?.TeamNames?.[1] || "Team 1 Agent 2";
+    const team2_text = info?.TeamNames?.[2] || "Team 2 Agent 1";
+    const team2_text_two = info?.TeamNames?.[3] || "Team 2 Agent 2";
 
     const ctx = canvas.getContext("2d");
     const padding = 20;
@@ -122,6 +124,12 @@ async function renderer(context) {
 
     line++;
 
+    // Team Row
+    ctx.fillText(team1_text_two, team1_x, line_height * line);
+    ctx.fillText(team2_text_two, team2_x, line_height * line);
+
+    line++;
+
     // Question Row
     ctx.fillText("Question:", label_x, question_y);
     let wrappedText1 = wrapText(ctx, team1_question, team1_x, question_y, row_width, line_height);

kaggle_environments/envs/llm_20_questions/llm_20_questions.py

@@ -31,6 +31,9 @@ ASK = "ask"
 GUESSER = "guesser"
 ANSWERER = "answerer"
 
+RANDOM_GUESSER = "random_guesser"
+RANDOM_ANSWERER = "random_answerer"
+
 keywords_list = json.loads(KEYWORDS_JSON)
 keyword_cat = random.choice(keywords_list)
 category = keyword_cat["category"]
@@ -39,6 +42,21 @@ keyword = keyword_obj["keyword"]
 alts = keyword_obj["alts"]
 
 
+def random_guesser(obs):
+    if obs.turnType == GUESS:
+        return "banana"
+    if random.random() < .5:
+        return "Is is a person?"
+    if random.random() < .5:
+        return "Is it a place?"
+    return "Is it a thing?"
+
+def random_answerer():
+    if random.random() > .5:
+        return "yes"
+    return "no"
+
+
 def guesser_agent(obs):
     info_prompt = """You are playing a game of 20 questions where you ask the questions and try to figure out the keyword, which will be a real or fictional person, place, or thing. \nHere is what you know so far:\n{q_a_thread}"""
     questions_prompt = """Ask one yes or no question."""
@@ -84,14 +102,16 @@ def answerer_agent(obs):
         return ""
 
 
-agents = {GUESSER: guesser_agent, ANSWERER: answerer_agent}
+agents = {GUESSER: guesser_agent, ANSWERER: answerer_agent, RANDOM_ANSWERER: random_answerer, RANDOM_GUESSER: random_guesser}
 
 def guesser_action(active, inactive, step):
     inactive.observation.keyword = keyword
     inactive.observation.category = category
     guessed = False
+    bad_guess = False
     if not active.action:
         active.status = ERROR
+        bad_guess = True
     elif active.observation.turnType == ASK:
         question = active.action[:2000]
         active.observation.questions.append(question)
@@ -105,7 +125,7 @@ def guesser_action(active, inactive, step):
         score = 20 - int(step / 3)
         end_game(active, score, DONE)
         end_game(inactive, score, DONE)
-    return guessed
+    return [guessed, bad_guess]
 
 def end_game(agent, reward, status):
     agent.observation.keyword = keyword
@@ -174,6 +194,8 @@ def interpreter(state, env):
     end_early = (active1 and active1.status) in (TIMEOUT, ERROR) or (active2 and active2.status in (TIMEOUT, ERROR))
     one_guessed = False
     two_guessed = False
+    one_bad_guess = False
+    two_bad_guess = False
     one_bad_response = False
     two_bad_response = False
 
@@ -181,28 +203,28 @@ def interpreter(state, env):
         raise ValueError
 
     if active1.observation.role == GUESSER:
-        one_guessed = guesser_action(active1, inactive1, step)
+        [one_guessed, one_bad_guess] = guesser_action(active1, inactive1, step)
     else:
         one_bad_response = answerer_action(active1, inactive1)
 
     if active2.observation.role == GUESSER:
-        two_guessed = guesser_action(active2, inactive2, step)
+        [two_guessed, two_bad_guess] = guesser_action(active2, inactive2, step)
     else:
         two_bad_response = answerer_action(active2, inactive2)
 
-    if active1.status in (TIMEOUT, ERROR) or one_bad_response:
+    if active1.status in (TIMEOUT, ERROR) or one_bad_response or one_bad_guess:
         end_game(active1, -1, active1.status)
         end_game(inactive1, 1, DONE)
-    elif end_early or two_bad_response:
+    elif end_early or two_bad_response or two_bad_guess:
         end_game(active1, 1, DONE)
         end_game(inactive1, 1, DONE)
     else:
         increment_turn(active1, inactive1, step, one_guessed)
     
-    if active2.status in (TIMEOUT, ERROR) or two_bad_response:
+    if active2.status in (TIMEOUT, ERROR) or two_bad_response or two_bad_guess:
         end_game(active2, -1, active2.status)
         end_game(inactive2, 1, DONE)
-    elif end_early or one_bad_response:
+    elif end_early or one_bad_response or one_bad_guess:
         end_game(active2, 1, DONE)
         end_game(inactive2, 1, DONE)
     else:

kaggle_environments/envs/llm_20_questions/test_llm_20_questions.py

@@ -6,6 +6,15 @@ def custom_questioner(obs):
         return "banana"
     return "Is it a banana?"
 
+def last_round_guesser_error(obs):
+    if obs.turnType == "guess" and len(obs.questions) == 20:
+         a = 1
+         b = 0
+         return a / b
+    if obs.turnType == "guess":
+        return "banana"
+    return "Is it a banana?"
+
 def custom_answerer():
     return "no"
 
@@ -39,7 +48,6 @@ def test_llm_20_q_errors_on_error_answer():
     assert json["name"] == "llm_20_questions"
     assert json["rewards"] == [1, 1, 1, None]
     assert json["statuses"] == ["DONE", "DONE", "DONE", "ERROR"]
-    print(len(json["steps"]))
     assert len(json["steps"]) == 3
 
 def test_llm_20_q_errors_on_error_question():
@@ -49,5 +57,12 @@ def test_llm_20_q_errors_on_error_question():
     assert json["name"] == "llm_20_questions"
     assert json["rewards"] == [1, 1, None, 1]
     assert json["statuses"] == ["DONE", "DONE", "ERROR", "DONE"]
-    print(len(json["steps"]))
-    assert len(json["steps"]) == 2
+    assert len(json["steps"]) == 2
+
+def test_llm_20_q_errors_on_error_last_guess():
+    env = make("llm_20_questions", debug=True)
+    env.run([custom_questioner, custom_answerer, last_round_guesser_error, custom_answerer])
+    json = env.toJSON()
+    assert json["name"] == "llm_20_questions"
+    assert json["rewards"] == [1, 1, None, 1]
+    assert json["statuses"] == ["DONE", "DONE", "ERROR", "DONE"]

{kaggle_environments-1.14.11.dist-info → kaggle_environments-1.14.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kaggle-environments
-Version: 1.14.11
+Version: 1.14.12
 Summary: Kaggle Environments
 Home-page: https://github.com/Kaggle/kaggle-environments
 Author: Kaggle

{kaggle_environments-1.14.11.dist-info → kaggle_environments-1.14.12.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kaggle_environments/__init__.py,sha256=jB8Zxnx8_IrtPX8ArsCgrfeRPXcejHOZvDx1-PhHiE0,1703
+kaggle_environments/__init__.py,sha256=oU4_4nptOEcuW1zMtVey6bzd7GOtU50VuXXIwrqijK0,1703
 kaggle_environments/agent.py,sha256=BpYrAC6EOhJF3Wzynq-CWg8l-z5uXW1Lq4qUsyOM0ik,6203
 kaggle_environments/api.py,sha256=eLBKqr11Ku4tdsMUdUqy74FIVEA_hdV3_QUpX84x3Z8,798
 kaggle_environments/core.py,sha256=IrEkN9cIA2djBAxI8Sz1GRpGNKjhqbnBdV6irAeTm8Q,27851
@@ -70,10 +70,10 @@ kaggle_environments/envs/kore_fleets/starter_bots/ts/test/configuration.json,sha
 kaggle_environments/envs/kore_fleets/starter_bots/ts/test/fullob.json,sha256=lEWzlv1rqiDMRmLqiGIO3j6m2Uf7OMjXx0zXBplAz1M,6791
 kaggle_environments/envs/kore_fleets/starter_bots/ts/test/observation.json,sha256=SIgLl8uTPmv2T_-g3gsFaibh_DWfvZNlZS7yPXL1Cnc,5998
 kaggle_environments/envs/llm_20_questions/keywords.py,sha256=BnyVrV0_xbiDt_Td4YXSKae9hHndAo5YYfYVtyL4eS4,45282
-kaggle_environments/envs/llm_20_questions/llm_20_questions.js,sha256=o30Nx1Tt0eAxBsQukmyxYlr7GuvptTUADWUnEnYDHe8,6009
+kaggle_environments/envs/llm_20_questions/llm_20_questions.js,sha256=3y_5q9c7EMaDJAbOnebK9xnzOh2z0eH4orW6eP1PMOs,6319
 kaggle_environments/envs/llm_20_questions/llm_20_questions.json,sha256=VYxgQEDCaJv910sQdE_FOtV2w1V83Y3C6nTjx4FFZyo,2101
-kaggle_environments/envs/llm_20_questions/llm_20_questions.py,sha256=Xj1slTw2WVwf94NNKV92qqJI52vuIY8quDwzuJq5Kgc,9563
-kaggle_environments/envs/llm_20_questions/test_llm_20_questions.py,sha256=sq04OePvY9VSR4Cjw4RI_i5FMwGni_ewpr5fcF2HldY,1831
+kaggle_environments/envs/llm_20_questions/llm_20_questions.py,sha256=kXRQfYQWCPE1rqkcHz3luebBnDKDuoW-Wk9aB2KLjIs,10236
+kaggle_environments/envs/llm_20_questions/test_llm_20_questions.py,sha256=GxVurD9rO8H78xFmG15HpqbyEFj4yoRd79MNX8UtxHo,2375
 kaggle_environments/envs/lux_ai_2021/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kaggle_environments/envs/lux_ai_2021/agents.py,sha256=bDqsY17Jxrk69a8HXb96ex2U1ipKkQPVJuXxmiA5hMA,421
 kaggle_environments/envs/lux_ai_2021/index.html,sha256=4Ec1FuHuuH1EvkvlLOpNUUFYLovnS_IA-rOOHEUjcGY,1550
@@ -177,9 +177,9 @@ kaggle_environments/envs/tictactoe/tictactoe.js,sha256=NZDT-oSG0a6a-rso9Ldh9qkJw
 kaggle_environments/envs/tictactoe/tictactoe.json,sha256=zMXZ8-fpT7FBhzz2FFBvRLn4XwtngjEqOieMvI6cCj8,1121
 kaggle_environments/envs/tictactoe/tictactoe.py,sha256=iLNU5V-lz7Xab-d1vpPMfU5jDM3QtgBUH63Y_SU7I9Y,3639
 kaggle_environments/static/player.html,sha256=HH8qvFfTIDw8eZvw5W88jcCu58Lo4_eUQ1ak46KWVGQ,22945
-kaggle_environments-1.14.11.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kaggle_environments-1.14.11.dist-info/METADATA,sha256=uL7fO1pEmq1BAPPyPUXv3mBa3JSlWnN8LbJcZb_kXfc,10701
-kaggle_environments-1.14.11.dist-info/WHEEL,sha256=m9WAupmBd2JGDsXWQGJgMGXIWbQY3F5c2xBJbBhq0nY,110
-kaggle_environments-1.14.11.dist-info/entry_points.txt,sha256=HbVC-LKGQFV6lEEYBYyDTtrkHgdHJUWQ8_qt9KHGqz4,70
-kaggle_environments-1.14.11.dist-info/top_level.txt,sha256=v3MMWIPMQFcI-WuF_dJngHWe9Bb2yH_6p4wat1x4gAc,20
-kaggle_environments-1.14.11.dist-info/RECORD,,
+kaggle_environments-1.14.12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kaggle_environments-1.14.12.dist-info/METADATA,sha256=nPl33KIBUPk78LAv-6FWRvFRRzSEdoqp5a5AUezMs4M,10701
+kaggle_environments-1.14.12.dist-info/WHEEL,sha256=m9WAupmBd2JGDsXWQGJgMGXIWbQY3F5c2xBJbBhq0nY,110
+kaggle_environments-1.14.12.dist-info/entry_points.txt,sha256=HbVC-LKGQFV6lEEYBYyDTtrkHgdHJUWQ8_qt9KHGqz4,70
+kaggle_environments-1.14.12.dist-info/top_level.txt,sha256=v3MMWIPMQFcI-WuF_dJngHWe9Bb2yH_6p4wat1x4gAc,20
+kaggle_environments-1.14.12.dist-info/RECORD,,