kaggle-environments 1.15.2__py2.py3-none-any.whl → 1.16.0__py2.py3-none-any.whl

kaggle_environments/envs/lux_ai_s3/test_agents/python/lux/kit.py

@@ -0,0 +1,27 @@
+import numpy as np
+def to_json(obj):
+    if isinstance(obj, np.ndarray):
+        return obj.tolist()
+    elif isinstance(obj, np.integer):
+        return int(obj)
+    elif isinstance(obj, np.floating):
+        return float(obj)
+    elif isinstance(obj, list) or isinstance(obj, tuple):
+        return [to_json(s) for s in obj]
+    elif isinstance(obj, dict):
+        out = {}
+        for k in obj:
+            out[k] = to_json(obj[k])
+        return out
+    else:
+        return obj
+def from_json(state):
+    if isinstance(state, list):
+        return np.array(state)
+    elif isinstance(state, dict):
+        out = {}
+        for k in state:
+            out[k] = from_json(state[k])
+        return out
+    else:
+        return state 

kaggle_environments/envs/lux_ai_s3/test_agents/python/lux/utils.py

@@ -0,0 +1,17 @@
+# direction (0 = center, 1 = up, 2 = right, 3 = down, 4 = left)
+def direction_to(src, target):
+    ds = target - src
+    dx = ds[0]
+    dy = ds[1]
+    if dx == 0 and dy == 0:
+        return 0
+    if abs(dx) > abs(dy):
+        if dx > 0:
+            return 2 
+        else:
+            return 4
+    else:
+        if dy > 0:
+            return 3
+        else:
+            return 1

kaggle_environments/envs/lux_ai_s3/test_agents/python/main.py

@@ -0,0 +1,53 @@
+import json
+import sys
+from argparse import Namespace
+if __package__ == "":
+    from agent import Agent
+    from lux.kit import from_json
+else:
+    from .agent import Agent
+    from .lux.kit import from_json
+### DO NOT REMOVE THE FOLLOWING CODE ###
+agent_dict = dict() # store potentially multiple dictionaries as kaggle imports code directly
+agent_prev_obs = dict()
+def agent_fn(observation, configurations):
+    """
+    agent definition for kaggle submission.
+    """
+    global agent_dict
+    obs = observation.obs
+    if type(obs) == str:
+        obs = json.loads(obs)
+    step = observation.step
+    player = observation.player
+    remainingOverageTime = observation.remainingOverageTime
+    if step == 0:
+        agent_dict[player] = Agent(player, configurations["env_cfg"])
+    agent = agent_dict[player]
+    actions = agent.act(step, from_json(obs), remainingOverageTime)
+    return dict(action=actions.tolist())
+if __name__ == "__main__":
+    
+    def read_input():
+        """
+        Reads input from stdin
+        """
+        try:
+            return input()
+        except EOFError as eof:
+            raise SystemExit(eof)
+    step = 0
+    player_id = 0
+    env_cfg = None
+    i = 0
+    while True:
+        inputs = read_input()
+        raw_input = json.loads(inputs)
+        observation = Namespace(**dict(step=raw_input["step"], obs=raw_input["obs"], remainingOverageTime=raw_input["remainingOverageTime"], player=raw_input["player"], info=raw_input["info"]))
+        if i == 0:
+            env_cfg = raw_input["info"]["env_cfg"]
+            player_id = raw_input["player"]
+        i += 1
+        actions = agent_fn(observation, dict(env_cfg=env_cfg))
+        # send actions to engine
+        print(json.dumps(actions))

kaggle_environments/envs/lux_ai_s3/test_lux.py

@@ -0,0 +1,9 @@
+import sys
+from kaggle_environments import make
+
+def xtest_lux_completes():
+    env = make("lux_ai_s3", debug=True)
+    env.run(["random_agent", "random_agent"])
+    json = env.toJSON()
+    assert json["name"] == "lux_ai_s3"
+    assert json["statuses"] == ["DONE", "DONE"]