kaggle-environments 1.16.5__tar.gz → 1.16.6__tar.gz

{kaggle-environments-1.16.5/kaggle_environments.egg-info → kaggle-environments-1.16.6}/PKG-INFO

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

{kaggle-environments-1.16.5 → kaggle-environments-1.16.6}/kaggle_environments/__init__.py

@@ -20,7 +20,7 @@ from .core import *
 from .main import http_request
 from . import errors
 
-__version__ = "1.16.5"
+__version__ = "1.16.6"
 
 __all__ = ["Agent", "environments", "errors", "evaluate", "http_request",
            "make", "register", "utils", "__version__",

{kaggle-environments-1.16.5 → kaggle-environments-1.16.6}/kaggle_environments/agent.py

@@ -79,7 +79,7 @@ class UrlAgent:
                 "observation": observation,
             },
         }
-        timeout = float(observation.remainingOverageTime) + float(configuration.actTimeout) + 10
+        timeout = float(observation.remainingOverageTime) + float(configuration.actTimeout) + 1
         try:
             response = requests.post(url=self.raw, data=json.dumps(data), timeout=timeout)
             response.raise_for_status()

kaggle-environments-1.16.6/kaggle_environments/envs/lux_ai_s3/lux_ai_s3.py

@@ -0,0 +1,141 @@
+import dataclasses
+import json
+import math
+import os
+import random
+import sys
+from os import path
+from .agents import all_agents
+from sys import path as syspath
+from os import path as osp
+
+# next two lines enables importing local packages e.g. luxai_s3
+__dir__ = osp.dirname(__file__)
+syspath.append(__dir__)
+import numpy as np
+
+import copy
+import json
+def to_json(state):
+    if isinstance(state, np.ndarray):
+        return state.tolist()
+    elif isinstance(state, np.int64):
+        return state.tolist()
+    elif isinstance(state, list):
+        return [to_json(s) for s in state]
+    elif isinstance(state, dict):
+        out = {}
+        for k in state:
+            out[k] = to_json(state[k])
+        return out
+    else:
+        return state
+prev_step = 0
+luxenv = None
+prev_obs = None
+state_obs = None
+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):
+    try:
+        from luxai_s3.wrappers import LuxAIS3GymEnv, RecordEpisode
+        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
+        # filter_actions(state, env)
+
+        if env.done:
+            if "seed" in env.configuration:
+                seed = int(env.configuration["seed"])
+            else:
+                seed = math.floor(random.random() * 1e9);
+                env.configuration["seed"] = seed
+
+            luxenv = LuxAIS3GymEnv(numpy_output=True)
+            luxenv = RecordEpisode(luxenv, save_on_close=False, save_on_reset=False)
+            obs, info = luxenv.reset(seed=seed)
+
+            env_cfg_json = info["params"]
+
+            env.configuration.env_cfg = env_cfg_json
+            
+            player_0.observation.player = "player_0"
+            player_1.observation.player = "player_1"
+            player_0.observation.obs = json.dumps(to_json(obs["player_0"]))
+            player_1.observation.obs = json.dumps(to_json(obs["player_1"]))
+            
+            replay_frame = luxenv.serialize_episode_data(dict(
+                states=[luxenv.episode["states"][-1]],
+                metadata=luxenv.episode["metadata"],
+                params=luxenv.episode["params"]
+            ))
+            # don't need to keep metadata/params beyond first step
+            player_0.info = dict(replay=replay_frame)
+            return state
+        
+        new_state_obs, rewards, terminations, truncations, infos = luxenv.step({
+            "player_0": np.array(player_0.action["action"]),
+            "player_1": np.array(player_1.action["action"])
+        })
+        
+        # cannot store np arrays in replay jsons so must convert to list
+        player_0.action = player_0.action["action"]
+        player_1.action = player_1.action["action"]
+        
+        dones = dict()
+        for k in terminations:
+            dones[k] = terminations[k] | truncations[k]
+
+        player_0.observation.player = "player_0"
+        player_1.observation.player = "player_1"
+
+        player_0.observation.obs = json.dumps(to_json(new_state_obs["player_0"]))
+        player_1.observation.obs = json.dumps(to_json(new_state_obs["player_1"]))
+        
+
+        player_0.reward = int(rewards["player_0"])
+        player_1.reward = int(rewards["player_1"])
+
+        player_0.observation.reward = int(player_0.reward)
+        player_1.observation.reward = int(player_1.reward)
+        replay_frame = luxenv.serialize_episode_data(dict(
+            states=[luxenv.episode["states"][-1]],
+            actions=[luxenv.episode["actions"][-1]],
+            metadata=luxenv.episode["metadata"],
+            params=luxenv.episode["params"]
+        ))
+        # don't need to keep metadata/params beyond first step
+        del replay_frame["metadata"]
+        del replay_frame["params"]
+        player_0.info = dict(replay=replay_frame)
+
+        if np.all([dones[k] for k in dones]):
+            if player_0.status == "ACTIVE":
+                player_0.status = "DONE"
+            if player_1.status == "ACTIVE":
+                player_1.status = "DONE"
+        return state
+    except ModuleNotFoundError as e:
+        print("Lux AI S3 Dependencies are missing, interpreter will not work")
+    return state
+
+def renderer(state, env):
+    raise NotImplementedError("To render the replay, please set the render mode to json or html")
+
+
+dir_path = path.dirname(__file__)
+json_path = path.abspath(path.join(dir_path, "lux_ai_s3.json"))
+with open(json_path) as json_file:
+    specification = json.load(json_file)
+
+
+def html_renderer():
+    html_path = path.abspath(path.join(dir_path, "index.html"))
+    return ("html_path", html_path)
+
+agents = all_agents

{kaggle-environments-1.16.5 → kaggle-environments-1.16.6/kaggle_environments.egg-info}/PKG-INFO

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

kaggle-environments-1.16.5/kaggle_environments/envs/lux_ai_s3/lux_ai_s3.py

@@ -1,137 +0,0 @@
-import dataclasses
-import json
-import math
-import os
-import random
-import sys
-from os import path
-from .agents import all_agents
-from sys import path as syspath
-from os import path as osp
-
-# next two lines enables importing local packages e.g. luxai_s3
-__dir__ = osp.dirname(__file__)
-syspath.append(__dir__)
-import numpy as np
-
-import copy
-import json
-def to_json(state):
-    if isinstance(state, np.ndarray):
-        return state.tolist()
-    elif isinstance(state, np.int64):
-        return state.tolist()
-    elif isinstance(state, list):
-        return [to_json(s) for s in state]
-    elif isinstance(state, dict):
-        out = {}
-        for k in state:
-            out[k] = to_json(state[k])
-        return out
-    else:
-        return state
-prev_step = 0
-luxenv = None
-prev_obs = None
-state_obs = None
-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):
-    from luxai_s3.wrappers import LuxAIS3GymEnv, RecordEpisode
-    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
-    # filter_actions(state, env)
-
-    if env.done:
-        if "seed" in env.configuration:
-            seed = int(env.configuration["seed"])
-        else:
-            seed = math.floor(random.random() * 1e9);
-            env.configuration["seed"] = seed
-
-        luxenv = LuxAIS3GymEnv(numpy_output=True)
-        luxenv = RecordEpisode(luxenv, save_on_close=False, save_on_reset=False)
-        obs, info = luxenv.reset(seed=seed)
-
-        env_cfg_json = info["params"]
-
-        env.configuration.env_cfg = env_cfg_json
-        
-        player_0.observation.player = "player_0"
-        player_1.observation.player = "player_1"
-        player_0.observation.obs = json.dumps(to_json(obs["player_0"]))
-        player_1.observation.obs = json.dumps(to_json(obs["player_1"]))
-        
-        replay_frame = luxenv.serialize_episode_data(dict(
-            states=[luxenv.episode["states"][-1]],
-            metadata=luxenv.episode["metadata"],
-            params=luxenv.episode["params"]
-        ))
-        # don't need to keep metadata/params beyond first step
-        player_0.info = dict(replay=replay_frame)
-        return state
-    
-    new_state_obs, rewards, terminations, truncations, infos = luxenv.step({
-        "player_0": np.array(player_0.action["action"]),
-        "player_1": np.array(player_1.action["action"])
-    })
-    
-    # cannot store np arrays in replay jsons so must convert to list
-    player_0.action = player_0.action["action"]
-    player_1.action = player_1.action["action"]
-    
-    dones = dict()
-    for k in terminations:
-        dones[k] = terminations[k] | truncations[k]
-
-    player_0.observation.player = "player_0"
-    player_1.observation.player = "player_1"
-
-    player_0.observation.obs = json.dumps(to_json(new_state_obs["player_0"]))
-    player_1.observation.obs = json.dumps(to_json(new_state_obs["player_1"]))
-    
-
-    player_0.reward = int(rewards["player_0"])
-    player_1.reward = int(rewards["player_1"])
-
-    player_0.observation.reward = int(player_0.reward)
-    player_1.observation.reward = int(player_1.reward)
-    replay_frame = luxenv.serialize_episode_data(dict(
-        states=[luxenv.episode["states"][-1]],
-        actions=[luxenv.episode["actions"][-1]],
-        metadata=luxenv.episode["metadata"],
-        params=luxenv.episode["params"]
-    ))
-    # don't need to keep metadata/params beyond first step
-    del replay_frame["metadata"]
-    del replay_frame["params"]
-    player_0.info = dict(replay=replay_frame)
-
-    if np.all([dones[k] for k in dones]):
-        if player_0.status == "ACTIVE":
-            player_0.status = "DONE"
-        if player_1.status == "ACTIVE":
-            player_1.status = "DONE"
-    return state
-
-def renderer(state, env):
-    raise NotImplementedError("To render the replay, please set the render mode to json or html")
-
-
-dir_path = path.dirname(__file__)
-json_path = path.abspath(path.join(dir_path, "lux_ai_s3.json"))
-with open(json_path) as json_file:
-    specification = json.load(json_file)
-
-
-def html_renderer():
-    html_path = path.abspath(path.join(dir_path, "index.html"))
-    return ("html_path", html_path)
-
-agents = all_agents