weiss-sim 0.1.2__cp312-cp312-win_amd64.whl

weiss_sim/__init__.py

@@ -0,0 +1,72 @@
+from __future__ import annotations
+
+import numpy as np
+
+from .weiss_sim import (
+    ACTION_SPACE_SIZE,
+    OBS_LEN,
+    PASS_ACTION_ID,
+    SPEC_HASH,
+    BatchOutDebug,
+    BatchOutMinimal,
+    EnvPool,
+    __version__,
+)
+from .rl import RlStep, pass_action_id_for_decision_kind, reset_rl, step_rl
+
+
+class EnvPoolBuffers:
+    """Preallocated numpy buffers for high-throughput stepping."""
+
+    def __init__(self, pool: EnvPool) -> None:
+        self.pool = pool
+        num_envs = pool.envs_len
+        self.out = BatchOutMinimal(num_envs)
+        self.obs = self.out.obs
+        self.masks = self.out.masks
+        self.rewards = self.out.rewards
+        self.terminated = self.out.terminated
+        self.truncated = self.out.truncated
+        self.actor = self.out.actor
+        self.decision_id = self.out.decision_id
+        self.engine_status = self.out.engine_status
+        self.spec_hash = self.out.spec_hash
+        self.legal_ids = np.empty(num_envs * pool.action_space, dtype=np.uint16)
+        self.legal_offsets = np.zeros(num_envs + 1, dtype=np.uint32)
+
+    def reset(self):
+        self.pool.reset_into(self.out)
+        return self.out
+
+    def reset_indices(self, indices):
+        self.pool.reset_indices_into(list(indices), self.out)
+        return self.out
+
+    def reset_done(self, done_mask):
+        self.pool.reset_done_into(done_mask, self.out)
+        return self.out
+
+    def step(self, actions):
+        self.pool.step_into(actions, self.out)
+        return self.out
+
+    def legal_action_ids(self):
+        count = self.pool.legal_action_ids_into(self.legal_ids, self.legal_offsets)
+        return self.legal_ids[:count], self.legal_offsets
+
+
+__all__ = [
+    "EnvPool",
+    "EnvPoolBuffers",
+    "BatchOutMinimal",
+    "BatchOutDebug",
+    "ACTION_SPACE_SIZE",
+    "OBS_LEN",
+    "SPEC_HASH",
+    "RlStep",
+    "reset_rl",
+    "step_rl",
+    "pass_action_id_for_decision_kind",
+    "PASS_ACTION_ID",
+    "__version__",
+]

weiss_sim/rl.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+import numpy as np
+
+from .weiss_sim import PASS_ACTION_ID, BatchOutMinimal, EnvPool
+
+
+@dataclass(frozen=True)
+class RlStep:
+    obs: np.ndarray
+    masks: np.ndarray
+    rewards: np.ndarray
+    terminated: np.ndarray
+    truncated: np.ndarray
+    actor: np.ndarray
+    decision_id: np.ndarray
+    engine_status: np.ndarray
+    spec_hash: np.ndarray
+
+
+def pass_action_id_for_decision_kind(decision_kind):
+    return PASS_ACTION_ID
+
+
+def reset_rl(pool: EnvPool) -> RlStep:
+    out = BatchOutMinimal(pool.envs_len)
+    pool.reset_into(out)
+    return RlStep(
+        obs=out.obs,
+        masks=out.masks,
+        rewards=out.rewards,
+        terminated=out.terminated,
+        truncated=out.truncated,
+        actor=out.actor,
+        decision_id=out.decision_id,
+        engine_status=out.engine_status,
+        spec_hash=out.spec_hash,
+    )
+
+
+def step_rl(pool: EnvPool, actions) -> RlStep:
+    out = BatchOutMinimal(pool.envs_len)
+    pool.step_into(actions, out)
+    return RlStep(
+        obs=out.obs,
+        masks=out.masks,
+        rewards=out.rewards,
+        terminated=out.terminated,
+        truncated=out.truncated,
+        actor=out.actor,
+        decision_id=out.decision_id,
+        engine_status=out.engine_status,
+        spec_hash=out.spec_hash,
+    )

weiss_sim-0.1.2.dist-info/METADATA

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: weiss-sim
+Version: 0.1.2
+Requires-Dist: numpy>=1.23
+Requires-Python: >=3.10

weiss_sim-0.1.2.dist-info/RECORD

@@ -0,0 +1,6 @@
+weiss_sim-0.1.2.dist-info/METADATA,sha256=SyRoMqTHL0cP1SHDFzr3cqrzijhWzNjnnKHCZKOm-go,104
+weiss_sim-0.1.2.dist-info/WHEEL,sha256=MicGETHZ2NSvTN4l4NAAzDXbOHnrlOKS0CnBsnFTiLs,97
+weiss_sim/__init__.py,sha256=1pcn5K4wCGxSycm-JmR0B5x3n2EZdXAOClvDwTafXDM,2025
+weiss_sim/rl.py,sha256=Ylh0NTOgf2XEubEw-LZNawNTzHZv5SRKMpXBISqTQAI,1395
+weiss_sim/weiss_sim.cp312-win_amd64.pyd,sha256=m2MnRg3tQuDmJsTEWVcu1la8nywe9AVGFIa2LYXwUoY,1466368
+weiss_sim-0.1.2.dist-info/RECORD,,

weiss_sim-0.1.2.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: maturin (1.10.2)
+Root-Is-Purelib: false
+Tag: cp312-cp312-win_amd64