kaggle-environments 1.14.3__py2.py3-none-any.whl → 1.14.5__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.3"
+__version__ = "1.14.5"
 
 __all__ = ["Agent", "environments", "errors", "evaluate", "http_request",
            "make", "register", "utils", "__version__",

kaggle_environments/envs/llm_20_questions/llm_20_questions.json

@@ -7,6 +7,7 @@
     "configuration": {
       "episodeSteps": 61,
       "actTimeout": 60,
+      "runTimeout": 9600,
       "agentTimeout": {
         "description": "Obsolete field kept for backwards compatibility, please use observation.remainingOverageTime.",
         "type": "number",
@@ -68,4 +69,4 @@
       "defaults": ["ACTIVE", "INACTIVE", "ACTIVE", "INACTIVE"]
     }
   }
-  
+  

kaggle_environments/envs/llm_20_questions/llm_20_questions.py

@@ -20,6 +20,17 @@ model = None
 tokenizer = None
 model_initialized = False
 
+ERROR = "ERROR"
+DONE = "DONE"
+INACTIVE = "INACTIVE"
+ACTIVE = "ACTIVE"
+TIMEOUT = "TIMEOUT"
+
+GUESS = "guess"
+ASK = "ask"
+GUESSER = "guesser"
+ANSWERER = "guesser"
+
 keywords_list = json.loads(KEYWORDS_JSON)
 keyword_cat = random.choice(keywords_list)
 category = keyword_cat["category"]
@@ -42,12 +53,12 @@ def guesser_agent(obs):
         )
 
     prompt = ""
-    if obs.turnType == "ask":
+    if obs.turnType == ASK:
         prompt = "{}{}".format(
             info_prompt.format(q_a_thread=q_a_thread),
             questions_prompt
         )
-    elif obs.turnType == "guess":
+    elif obs.turnType == GUESS:
         prompt = "{}{}".format(
             info_prompt.format(q_a_thread=q_a_thread),
             guess_prompt
@@ -73,7 +84,66 @@ def answerer_agent(obs):
         return ""
 
 
-agents = {"guesser": guesser_agent, "answerer": answerer_agent}
+agents = {GUESSER: guesser_agent, ANSWERER: answerer_agent}
+
+def guesser_action(active, inactive, step):
+    guessed = False
+    if not active.action:
+        active.status = ERROR
+    elif active.observation.turnType == ASK:
+        question = active.action[:2000]
+        active.observation.questions.append(question)
+        inactive.observation.questions.append(question)
+    elif active.observation.turnType == GUESS:
+        guess = active.action[:100]
+        active.observation.guesses.append(guess)
+        inactive.observation.guesses.append(guess)
+    if active.action and keyword_guessed(active.action):
+        guessed = True
+        score = 20 - int(step / 3)
+        end_game(active, inactive, score, DONE, DONE)
+    return guessed
+
+def end_game(active, inactive, reward, status, inactive_status):
+    active.observation.keyword = keyword
+    active.observation.category = category
+    inactive.observation.keyword = keyword
+    inactive.observation.category = category
+    active.reward = reward
+    inactive.reward = reward
+    active.status = status
+    inactive.status = inactive_status
+
+
+def answerer_action(active, inactive):
+    active.observation.keyword = keyword
+    active.observation.category = category
+    response = active.action
+    if not response:
+        response = "none"
+        end_game(active, inactive, -1, ERROR, DONE)
+    elif "yes" in response.lower():
+        response = "yes"
+    elif "no" in response.lower():
+        response = "no"
+    else:
+        response = "maybe"
+        end_game(active, inactive, -1, ERROR, DONE)
+    active.observation.answers.append(response)
+    inactive.observation.answers.append(response)
+
+def increment_turn(active, inactive, step, guessed):
+    if step == 59 and not guessed:
+        end_game(active, inactive, -1, DONE, DONE)
+    elif active.observation.turnType == "guess":
+        active.observation.turnType = "ask"
+    elif active.observation.turnType == "ask":
+        active.observation.turnType = "guess"
+        active.status = INACTIVE
+        inactive.status = ACTIVE
+    else:
+        active.status = INACTIVE
+        inactive.status = ACTIVE
 
 
 def interpreter(state, env):
@@ -81,14 +151,14 @@ def interpreter(state, env):
         return state
 
     # Isolate the active and inactive agents.
-    active1 = state[0] if state[0].status == "ACTIVE" else state[1]
-    inactive1 = state[0] if state[0].status == "INACTIVE" else state[1]
-    active2 = state[2] if state[2].status == "ACTIVE" else state[3]
-    inactive2 = state[2] if state[2].status == "INACTIVE" else state[3]
-    if active1.status == "DONE" and inactive1.status == "DONE":
+    active1 = state[0] if state[0].status == ACTIVE else state[1]
+    inactive1 = state[0] if state[0].status == INACTIVE else state[1]
+    active2 = state[2] if state[2].status == ACTIVE else state[3]
+    inactive2 = state[2] if state[2].status == INACTIVE else state[3]
+    if active1.status == DONE and inactive1.status == DONE:
         active1 = None
         inactive1 = None
-    if active2.status == "DONE" or inactive2.status == "DONE":
+    if active2.status == DONE or inactive2.status == DONE:
         active2 = None
         inactive2 = None
     if active1 is None and inactive1 is None and active2 is None and inactive2 is None:
@@ -96,110 +166,37 @@ def interpreter(state, env):
 
     step = state[0].observation.step
 
+    end_early = (active1 and active1.status) in (TIMEOUT, ERROR) or (active2 and active2.status in (TIMEOUT, ERROR))
+    either_guessed = False
+
     if active1 is not None:
         guessed = False
-        if active1.observation.role == "guesser":
-            if active1.observation.turnType == "ask":
-                active1.observation.questions.append(active1.action)
-                inactive1.observation.questions.append(active1.action)
-            elif active1.observation.turnType == "guess":
-                active1.observation.guesses.append(active1.action)
-                inactive1.observation.guesses.append(active1.action)
-            if keyword_guessed(active1.action):
-                guessed = True
-                score = 20 - int(step / 3)
-                active1.reward = score
-                inactive1.reward = score
-                active1.status = "DONE"
-                inactive1.status = "DONE"
-                active1.observation.keyword = keyword
-                active1.observation.category = category
-            inactive1.observation.keyword = keyword
-            inactive1.observation.category = category
+        if active1.observation.role == GUESSER:
+            guessed = guesser_action(active1, inactive1, step)
+            either_guessed = guessed
         else:
-            active1.observation.keyword = keyword
-            active1.observation.category = category
-            response = active1.action
-            if response.lower().__contains__("yes"):
-                response = "yes"
-            elif response.lower().__contains__("no"):
-                response = "no"
-            else:
-                response = "maybe"
-            active1.observation.answers.append(response)
-            inactive1.observation.answers.append(response)
-
-        if step == 59 and not guessed:
-            active1.observation.keyword = keyword
-            active1.observation.category = category
-            inactive1.observation.keyword = keyword
-            inactive1.observation.category = category
-            active1.reward = -1
-            inactive1.reward = -1
-            active1.status = "DONE"
-            inactive1.status = "DONE"
-        elif active1.observation.turnType == "guess":
-            active1.observation.turnType = "ask"
-        elif active1.observation.turnType == "ask":
-            active1.observation.turnType = "guess"
-            active1.status = "INACTIVE"
-            inactive1.status = "ACTIVE"
+            answerer_action(active1, inactive1)
+        if active1.status in (TIMEOUT, ERROR):
+            end_game(active1, inactive1, 0, active1.status, DONE)
+        elif end_early:
+            end_game(active1, inactive1, 0, DONE, DONE)
         else:
-            active1.status = "INACTIVE"
-            inactive1.status = "ACTIVE"
+            increment_turn(active1, inactive1, step, guessed)
     
     if active2 is not None:
         guessed = False
-        if active2.observation.role == "guesser":
-            if active2.observation.turnType == "ask":
-                active2.observation.questions.append(active2.action)
-                inactive2.observation.questions.append(active2.action)
-            elif active2.observation.turnType == "guess":
-                active2.observation.guesses.append(active2.action)
-                inactive2.observation.guesses.append(active2.action)
-            if keyword_guessed(active2.action):
-                guessed = True
-                score = 20 - int(step / 3)
-                active2.reward = score
-                inactive2.reward = score
-                active2.status = "DONE"
-                inactive2.status = "DONE"
-                active2.observation.keyword = keyword
-                active2.observation.category = category
-            inactive2.observation.keyword = keyword
-            inactive2.observation.category = category
+        if active2.observation.role == GUESSER:
+            guessed = guesser_action(active2, inactive2, step)
+            either_guessed = either_guessed or guessed
         else:
-            active2.observation.keyword = keyword
-            active2.observation.category = category
-            response = active2.action
-            if response.lower().__contains__("yes"):
-                response = "yes"
-            elif response.lower().__contains__("no"):
-                response = "no"
-            else:
-                response = "maybe"
-            active2.observation.answers.append(response)
-            inactive2.observation.answers.append(response)
-
-        if step == 59 and not guessed:
-            active2.observation.keyword = keyword
-            active2.observation.category = category
-            inactive2.observation.keyword = keyword
-            inactive2.observation.category = category
-            active2.reward = -1
-            inactive2.reward = -1
-            active2.status = "DONE"
-            inactive2.status = "DONE"
-        elif active2.observation.turnType == "guess":
-            active2.observation.turnType = "ask"
-        elif active2.observation.turnType == "ask":
-            active2.observation.turnType = "guess"
-            active2.status = "INACTIVE"
-            inactive2.status = "ACTIVE"
+            answerer_action(active2, inactive2)
+        if active2.status in (TIMEOUT, ERROR):
+            end_game(active2, inactive2, 0, active2.status, DONE)
+        elif end_early:
+            end_game(active2, inactive2, 0, DONE, DONE)
         else:
-            active2.status = "INACTIVE"
-            inactive2.status = "ACTIVE"
-
+            increment_turn(active2, inactive2, step, guessed)
+    
     return state
 
 
@@ -207,7 +204,7 @@ def renderer(state, env):
 
     for s in state:
         print("role: ", s.observation.role)
-        if s.observation.role == "guesser":
+        if s.observation.role == GUESSER:
             transcript = ""
             for i in range(0, len(s.observation.guesses)):
                 transcript = "{}Q: {} A: {}\nG: {}\n".format(

{kaggle_environments-1.14.3.dist-info → kaggle_environments-1.14.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kaggle-environments
-Version: 1.14.3
+Version: 1.14.5
 Summary: Kaggle Environments
 Home-page: https://github.com/Kaggle/kaggle-environments
 Author: Kaggle
@@ -10,17 +10,17 @@ Keywords: Kaggle
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: Flask (>=1.1.2)
-Requires-Dist: gymnasium (==0.29.0)
-Requires-Dist: jsonschema (>=3.0.1)
-Requires-Dist: numpy (>=1.19.5)
-Requires-Dist: pettingzoo (==1.24.0)
-Requires-Dist: requests (>=2.25.1)
-Requires-Dist: scipy (>=1.11.2)
-Requires-Dist: shimmy (>=1.2.1)
-Requires-Dist: stable-baselines3 (==2.1.0)
-Requires-Dist: transformers (>=4.33.1)
-Requires-Dist: vec-noise (>=1.1.4)
+Requires-Dist: Flask >=1.1.2
+Requires-Dist: gymnasium ==0.29.0
+Requires-Dist: jsonschema >=3.0.1
+Requires-Dist: numpy >=1.19.5
+Requires-Dist: pettingzoo ==1.24.0
+Requires-Dist: requests >=2.25.1
+Requires-Dist: scipy >=1.11.2
+Requires-Dist: shimmy >=1.2.1
+Requires-Dist: stable-baselines3 ==2.1.0
+Requires-Dist: transformers >=4.33.1
+Requires-Dist: vec-noise >=1.1.4
 
 # [<img src="https://kaggle.com/static/images/site-logo.png" height="50" style="margin-bottom:-15px" />](https://kaggle.com) Environments
 

{kaggle_environments-1.14.3.dist-info → kaggle_environments-1.14.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kaggle_environments/__init__.py,sha256=nmRV4RENpsoopKO_dSA9WfbSOfTQ0JD4tgIhpz1HfJw,1702
+kaggle_environments/__init__.py,sha256=sQ-D4OKiM-2xFoo9MlUPMX_TYBykHXx-VQE9QWp-ro0,1702
 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
@@ -71,8 +71,8 @@ kaggle_environments/envs/kore_fleets/starter_bots/ts/test/fullob.json,sha256=lEW
 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=qy6V_j7GOqjCbCowBdJ6dtr23HCvK2MdSbQWi-0blPY,5923
-kaggle_environments/envs/llm_20_questions/llm_20_questions.json,sha256=nFAPb4vc10qW3-WOOM-iEHtOro-ij7-Um4UdzRoK8Xc,2074
-kaggle_environments/envs/llm_20_questions/llm_20_questions.py,sha256=39ud8BnfwO9lyGViGQoKmRs3TUnqIR4EYbr0YjFC1y0,10141
+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=x5vPzxe0t-OZSUf_KBaiIlvW_PCbpKPqOE7ORPTCjXA,9073
 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
@@ -176,9 +176,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.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kaggle_environments-1.14.3.dist-info/METADATA,sha256=oEgT1xFfbYa8xb-XRdMhdUNfehjJvN6POkNyMX-k4HA,10722
-kaggle_environments-1.14.3.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-kaggle_environments-1.14.3.dist-info/entry_points.txt,sha256=HbVC-LKGQFV6lEEYBYyDTtrkHgdHJUWQ8_qt9KHGqz4,70
-kaggle_environments-1.14.3.dist-info/top_level.txt,sha256=v3MMWIPMQFcI-WuF_dJngHWe9Bb2yH_6p4wat1x4gAc,20
-kaggle_environments-1.14.3.dist-info/RECORD,,
+kaggle_environments-1.14.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kaggle_environments-1.14.5.dist-info/METADATA,sha256=Zn6TzB6J1NEXFrMd2W2M_DFkdXgqXQ_wZhx5uus8H18,10700
+kaggle_environments-1.14.5.dist-info/WHEEL,sha256=m9WAupmBd2JGDsXWQGJgMGXIWbQY3F5c2xBJbBhq0nY,110
+kaggle_environments-1.14.5.dist-info/entry_points.txt,sha256=HbVC-LKGQFV6lEEYBYyDTtrkHgdHJUWQ8_qt9KHGqz4,70
+kaggle_environments-1.14.5.dist-info/top_level.txt,sha256=v3MMWIPMQFcI-WuF_dJngHWe9Bb2yH_6p4wat1x4gAc,20
+kaggle_environments-1.14.5.dist-info/RECORD,,

{kaggle_environments-1.14.3.dist-info → kaggle_environments-1.14.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.38.4)
+Generator: bdist_wheel (0.41.1)
 Root-Is-Purelib: true
 Tag: py2-none-any
 Tag: py3-none-any