kaggle-environments 1.14.1__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.1"
+__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/envs/lux_ai_s2/lux_ai_s2.py

@@ -39,16 +39,13 @@ prev_step = 0
 luxenv: LuxAI_S2 = LuxAI_S2(verbose=0, validate_action_space=True, collect_stats=True)
 prev_obs = None
 state_obs = None
-def cleanup_dimensions():
-    global dimension_process
-    if dimension_process is not None:
-        dimension_process.kill()
+default_env_cfg = None
 def enqueue_output(out, queue):
     for line in iter(out.readline, b''):
         queue.put(line)
     out.close()
 def interpreter(state, env):
-    global luxenv, prev_obs, state_obs
+    global luxenv, prev_obs, state_obs, default_env_cfg
     player_0 = state[0]
     player_1 = state[1]
     # filter out actions such as debug annotations so they aren't saved
@@ -65,18 +62,29 @@ def interpreter(state, env):
         else:
             max_episode_length = 1000
 
-        parsed_env_config = copy.deepcopy(env.configuration)
-        parsed_env_config["max_episode_length"] = max_episode_length
-        delete_keys = ["seed", "episodeSteps", "actTimeout", "runTimeout", "env_cfg"]
         
-        for k in delete_keys:
-            if k in parsed_env_config: del parsed_env_config[k]
+        if default_env_cfg is None:
+            # if this is the first time creating the environment, env.configuration contains the kaggle competition configurations used to override 
+            # the default env config in LuxAI_S2. env.configuration is later populated by the merge of the kaggle competition configs with the 
+            # env config in LuxAI_S2 so we only run this branch once and save the result.
+            parsed_env_config = copy.deepcopy(env.configuration)
+            parsed_env_config["max_episode_length"] = max_episode_length
+            delete_keys = ["seed", "episodeSteps", "actTimeout", "runTimeout", "env_cfg"]
+            env_cfg_override = dict()
+            if "env_cfg" in parsed_env_config:
+                env_cfg_override = copy.deepcopy(parsed_env_config["env_cfg"])
+            for k in delete_keys:
+                if k in parsed_env_config: del parsed_env_config[k]
+            parsed_env_config = {**parsed_env_config, **env_cfg_override}
+            default_env_cfg = parsed_env_config
+        else:
+            parsed_env_config = default_env_cfg
         luxenv = LuxAI_S2(validate_action_space=True, collect_stats=True, **parsed_env_config)
         _, _ = luxenv.reset(seed=seed)
         state_obs = luxenv.state.get_compressed_obs()
 
         env_cfg_json = dataclasses.asdict(luxenv.env_cfg)
-        # del env_cfg_json["WEATHER_ID_TO_NAME"]
+
         env.configuration.env_cfg = env_cfg_json
         
         player_0.observation.player = "player_0"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kaggle-environments
-Version: 1.14.1
+Version: 1.14.5
 Summary: Kaggle Environments
 Home-page: https://github.com/Kaggle/kaggle-environments
 Author: Kaggle
@@ -16,7 +16,10 @@ 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.1.dist-info → kaggle_environments-1.14.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kaggle_environments/__init__.py,sha256=W-w3qHRxvExrwKAq0b95x02Ln4P18hiWeRHehZ_wSfw,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
@@ -105,7 +105,7 @@ kaggle_environments/envs/lux_ai_s2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQe
 kaggle_environments/envs/lux_ai_s2/agents.py,sha256=43bT4pa6t1SsAS4akQ8y2L7wJrs0hZAH0v1M5Nt_vR0,112
 kaggle_environments/envs/lux_ai_s2/index.html,sha256=-mAWnS23XMRMqDbT4Ceb7aFeFisKM12ri24GZBLNiKw,1606
 kaggle_environments/envs/lux_ai_s2/lux_ai_s2.json,sha256=eiiNUsvePpW2FTOVdJ4Lxy6P82TjOmc-6kA9V9Aqxks,1924
-kaggle_environments/envs/lux_ai_s2/lux_ai_s2.py,sha256=pUpFULFEZ-tsmVnEyTI9m_OKLdRxDtIbyFJ1xlGov5g,4555
+kaggle_environments/envs/lux_ai_s2/lux_ai_s2.py,sha256=R2CPS2yr2SsHzDtuAUA3k5evzGnibwNbSdwg5OZOwKg,5208
 kaggle_environments/envs/lux_ai_s2/test_lux.py,sha256=ngPPwEBbLZXpUkC0A6G8RdrCRW83yp8GmWJKuWusNyo,300
 kaggle_environments/envs/lux_ai_s2/luxai_s2/__init__.py,sha256=RR5qH3QiHJ_WxLf1iVi5bEPbFJYtySJ9Cpd5fhD-Dhk,59
 kaggle_environments/envs/lux_ai_s2/luxai_s2/actions.py,sha256=FtVPwW1muusolkpf9HbwNyQlCkeF96UAguWrjquwnuk,13446
@@ -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.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kaggle_environments-1.14.1.dist-info/METADATA,sha256=yRR4WK5x1Q6A4sSCBL9Tmit31aYhYjES6t48snhTnuE,10603
-kaggle_environments-1.14.1.dist-info/WHEEL,sha256=m9WAupmBd2JGDsXWQGJgMGXIWbQY3F5c2xBJbBhq0nY,110
-kaggle_environments-1.14.1.dist-info/entry_points.txt,sha256=HbVC-LKGQFV6lEEYBYyDTtrkHgdHJUWQ8_qt9KHGqz4,70
-kaggle_environments-1.14.1.dist-info/top_level.txt,sha256=v3MMWIPMQFcI-WuF_dJngHWe9Bb2yH_6p4wat1x4gAc,20
-kaggle_environments-1.14.1.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,,