jax-envelope 0.3.0__tar.gz → 0.4.0__tar.gz

{jax_envelope-0.3.0 → jax_envelope-0.4.0}/.github/workflows/publish.yml

@@ -28,7 +28,7 @@ jobs:
           uv sync --group dev --locked
 
       - name: Run tests
-        run: uv run pytest -m "not compat"
+        run: uv run pytest -m "not adapters"
 
   build:
     name: Build distribution

{jax_envelope-0.3.0 → jax_envelope-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jax-envelope
-Version: 0.3.0
+Version: 0.4.0
 Summary: A JAX-native environment interface with powerful wrappers and adapters for popular RL environment suites
 Project-URL: Homepage, https://github.com/keraJLi/envelope
 Project-URL: Repository, https://github.com/keraJLi/envelope
@@ -42,6 +42,7 @@ env = envelope.wrappers.ObservationNormalizationWrapper(env)
 ```
 
 ## 🌍 Simple, expressive interaction!
+
 - **Environments are pytrees**. Squish them through JAX transformations and trace their parameters.
 - **Idiomatic jax-y interface** of `init(key: Key) -> State, Info` and `step(state: State, action: PyTree) -> State, Info`. You can directly `jax.scan` over a `step(...)`!
 - **Spaces are super simple**. No `Tuple`, `Dict` nonsense! There are two spaces: `Continuous` and `Discrete`, which you can compose into a `PyTreeSpace`.
@@ -53,8 +54,6 @@ env = envelope.wrappers.ObservationNormalizationWrapper(env)
 - **Carry state across episodes** to track running statistics, for example to normalize observations.
 - **Composable wrappers** can be stacked in any order. For example, `ObservationNormalizationWrapper` before vs. after `VmapWrapper` gives per-env vs. global normalization.
 
-
-
 ## 🔌 Adapters for existing suites
 
 
@@ -78,11 +77,11 @@ let's you create environments from any of the above!
 
 ## 📝 Testing
 
-- **Default (no optional compat deps required)**: `uv run pytest -m "not compat"`
-- **Compat suite (requires full compat dependency group)**:
-  - `uv sync --group compat`
-  - `uv run pytest -m compat`
-  - If any compat dependency is missing/broken, the run will fail fast with an error telling you what to install.
+- **Default (no optional adapters deps required)**: `uv run pytest -m "not adapters"`
+- **Adapters suite (requires full adapters dependency group)**:
+  - `uv sync --group adapters`
+  - `uv run pytest -m adapters`
+  - If any adapter dependency is missing/broken, the run will fail fast with an error telling you what to install.
 
 ## 🏗️ Installation
 

{jax_envelope-0.3.0 → jax_envelope-0.4.0}/README.md

@@ -16,6 +16,7 @@ env = envelope.wrappers.ObservationNormalizationWrapper(env)
 ```
 
 ## 🌍 Simple, expressive interaction!
+
 - **Environments are pytrees**. Squish them through JAX transformations and trace their parameters.
 - **Idiomatic jax-y interface** of `init(key: Key) -> State, Info` and `step(state: State, action: PyTree) -> State, Info`. You can directly `jax.scan` over a `step(...)`!
 - **Spaces are super simple**. No `Tuple`, `Dict` nonsense! There are two spaces: `Continuous` and `Discrete`, which you can compose into a `PyTreeSpace`.
@@ -27,8 +28,6 @@ env = envelope.wrappers.ObservationNormalizationWrapper(env)
 - **Carry state across episodes** to track running statistics, for example to normalize observations.
 - **Composable wrappers** can be stacked in any order. For example, `ObservationNormalizationWrapper` before vs. after `VmapWrapper` gives per-env vs. global normalization.
 
-
-
 ## 🔌 Adapters for existing suites
 
 
@@ -52,11 +51,11 @@ let's you create environments from any of the above!
 
 ## 📝 Testing
 
-- **Default (no optional compat deps required)**: `uv run pytest -m "not compat"`
-- **Compat suite (requires full compat dependency group)**:
-  - `uv sync --group compat`
-  - `uv run pytest -m compat`
-  - If any compat dependency is missing/broken, the run will fail fast with an error telling you what to install.
+- **Default (no optional adapters deps required)**: `uv run pytest -m "not adapters"`
+- **Adapters suite (requires full adapters dependency group)**:
+  - `uv sync --group adapters`
+  - `uv run pytest -m adapters`
+  - If any adapter dependency is missing/broken, the run will fail fast with an error telling you what to install.
 
 ## 🏗️ Installation
 

{jax_envelope-0.3.0 → jax_envelope-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "jax-envelope"
-version = "0.3.0"
+version = "0.4.0"
 description = "A JAX-native environment interface with powerful wrappers and adapters for popular RL environment suites"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -52,7 +52,7 @@ allow-direct-references = true
 packages = ["src/envelope"]
 
 [dependency-groups]
-compat = [
+adapters = [
     "brax>=0.13.0",
     "craftax>=1.4.3",
     "navix>=0.7.0",
@@ -80,5 +80,5 @@ kinetix-env = { git = "https://github.com/FLAIROx/Kinetix.git" }
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 markers = [
-  "compat: tests requiring optional compat dependencies",
+  "adapters: tests requiring optional adapters dependencies",
 ]

{jax_envelope-0.3.0 → jax_envelope-0.4.0}/src/envelope/__init__.py

@@ -1,4 +1,4 @@
-from envelope.compat import create
+from envelope.adapters import create
 from envelope.environment import Environment, Info, InfoContainer
 from envelope.spaces import BatchedSpace, Continuous, Discrete, PyTreeSpace, Space
 from envelope.struct import Container, FrozenPyTreeNode, field, static_field

{jax_envelope-0.3.0/src/envelope/compat → jax_envelope-0.4.0/src/envelope/adapters}/__init__.py

@@ -4,14 +4,14 @@ from typing import Any, Protocol, Self
 
 # Lazy imports to avoid requiring all dependencies at once
 _env_module_map = {
-    "gymnax": ("envelope.compat.gymnax_envelope", "GymnaxEnvelope"),
-    "brax": ("envelope.compat.brax_envelope", "BraxEnvelope"),
-    "navix": ("envelope.compat.navix_envelope", "NavixEnvelope"),
-    "jumanji": ("envelope.compat.jumanji_envelope", "JumanjiEnvelope"),
-    "kinetix": ("envelope.compat.kinetix_envelope", "KinetixEnvelope"),
-    "craftax": ("envelope.compat.craftax_envelope", "CraftaxEnvelope"),
+    "gymnax": ("envelope.adapters.gymnax_envelope", "GymnaxEnvelope"),
+    "brax": ("envelope.adapters.brax_envelope", "BraxEnvelope"),
+    "navix": ("envelope.adapters.navix_envelope", "NavixEnvelope"),
+    "jumanji": ("envelope.adapters.jumanji_envelope", "JumanjiEnvelope"),
+    "kinetix": ("envelope.adapters.kinetix_envelope", "KinetixEnvelope"),
+    "craftax": ("envelope.adapters.craftax_envelope", "CraftaxEnvelope"),
     "mujoco_playground": (
-        "envelope.compat.mujoco_playground_envelope",
+        "envelope.adapters.mujoco_playground_envelope",
         "MujocoPlaygroundEnvelope",
     ),
 }

{jax_envelope-0.3.0/src/envelope/compat → jax_envelope-0.4.0/src/envelope/adapters}/craftax_envelope.py

@@ -10,7 +10,7 @@ from craftax.craftax_classic.envs.craftax_state import (
 from craftax.craftax_env import make_craftax_env_from_name
 
 from envelope import spaces as envelope_spaces
-from envelope.compat.gymnax_envelope import _convert_space as _convert_gymnax_space
+from envelope.adapters.gymnax_envelope import _convert_space as _convert_gymnax_space
 from envelope.environment import Environment, Info, InfoContainer, State
 from envelope.struct import Container, static_field
 from envelope.typing import Key, PyTree, TypeAlias

{jax_envelope-0.3.0/src/envelope/compat → jax_envelope-0.4.0/src/envelope/adapters}/kinetix_envelope.py

@@ -1,7 +1,7 @@
 """Kinetix compatibility wrapper.
 
 This module exposes Kinetix environments through the `envelope.environment.Environment`
-API. It mirrors envelope's compat philosophy:
+API. It mirrors envelope's adapters philosophy:
 - prefer *no* environment-side auto-reset (use `AutoResetWrapper` in envelope)
 - prefer *no* fixed episode time-limits (use `TruncationWrapper` in envelope)
 
@@ -30,7 +30,7 @@ from kinetix.util.saving import load_from_json_file
 
 from envelope import field
 from envelope import spaces as envelope_spaces
-from envelope.compat.gymnax_envelope import _convert_space as _convert_gymnax_space
+from envelope.adapters.gymnax_envelope import _convert_space as _convert_gymnax_space
 from envelope.environment import Environment, Info, InfoContainer, State
 from envelope.struct import Container, static_field
 from envelope.typing import Key, PyTree

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/conftest.py

@@ -1,6 +1,6 @@
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 
 @pytest.fixture(scope="module")

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/contract.py

@@ -1,6 +1,6 @@
-"""Shared contract helpers for compat wrappers.
+"""Shared contract helpers for adapters.
 
-These functions enforce a consistent baseline across all compat wrappers:
+These functions enforce a consistent baseline across all adapters:
 - reset/step return (state, info) with Info fields present
 - reward is scalar-ish and finite
 - action sampling is valid for action_space

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_brax_compat.py

@@ -1,4 +1,4 @@
-"""Tests for envelope.compat.brax_envelope module."""
+"""Tests for envelope.adapters.brax_envelope module."""
 
 # ruff: noqa: E402
 
@@ -7,14 +7,14 @@ from copy import deepcopy
 import jax
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 pytest.importorskip("brax")
 
 from brax.envs import Wrapper as BraxWrapper
 
-from envelope.compat.brax_envelope import BraxEnvelope
-from tests.compat.contract import (
+from envelope.adapters.brax_envelope import BraxEnvelope
+from tests.adapters.contract import (
     assert_jitted_rollout_contract,
     assert_reset_step_contract,
 )

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_craftax_compat.py

@@ -1,4 +1,4 @@
-"""Tests for envelope.compat.craftax_envelope module."""
+"""Tests for envelope.adapters.craftax_envelope module."""
 
 # ruff: noqa: E402
 
@@ -8,12 +8,12 @@ import jax
 import jax.numpy as jnp
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 pytest.importorskip("craftax")
 
 from envelope.spaces import Continuous, Discrete
-from tests.compat.contract import (
+from tests.adapters.contract import (
     assert_jitted_rollout_contract,
     assert_reset_step_contract,
 )
@@ -35,7 +35,7 @@ def craftax_env_id(request: pytest.FixtureRequest) -> str:
 
 @pytest.fixture(scope="module")
 def craftax_env(craftax_env_id: str):
-    from envelope.compat.craftax_envelope import CraftaxEnvelope
+    from envelope.adapters.craftax_envelope import CraftaxEnvelope
 
     return CraftaxEnvelope.from_name(craftax_env_id)
 
@@ -114,7 +114,7 @@ class _DummyEnv:
 
 
 def test_from_name_errors_on_auto_reset():
-    from envelope.compat.craftax_envelope import CraftaxEnvelope
+    from envelope.adapters.craftax_envelope import CraftaxEnvelope
 
     with pytest.raises(ValueError, match="Cannot override 'auto_reset' directly"):
         CraftaxEnvelope.from_name("AnyEnv", env_kwargs={"auto_reset": True})

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_create.py

@@ -1,4 +1,4 @@
-"""Unit tests for envelope.compat.create() factory function.
+"""Unit tests for envelope.adapters.create() factory function.
 
 These tests are dependency-free (no brax/gymnax/navix imports) and focus on:
 - parsing/validation of the env id
@@ -12,8 +12,8 @@ import types
 
 import pytest
 
-import envelope.compat as compat
-from envelope.compat import create
+import envelope.adapters as adapters
+from envelope.adapters import create
 
 
 def _install_dummy_suite(
@@ -44,7 +44,7 @@ def _install_dummy_suite(
             raise AssertionError(f"Unexpected import: {name}")
         return dummy_module
 
-    monkeypatch.setattr(compat, "_env_module_map", {suite: (module_name, class_name)})
+    monkeypatch.setattr(adapters, "_env_module_map", {suite: (module_name, class_name)})
     monkeypatch.setattr(importlib, "import_module", fake_import_module)
 
     return import_calls, from_name_calls
@@ -77,7 +77,7 @@ def test_create_unknown_suite_mentions_available_suites(
 ):
     # Keep the map deterministic so we can assert it appears in the message.
     monkeypatch.setattr(
-        compat, "_env_module_map", {"dummy": ("dummy_mod", "DummyWrapper")}
+        adapters, "_env_module_map", {"dummy": ("dummy_mod", "DummyWrapper")}
     )
 
     with pytest.raises(ValueError) as excinfo:
@@ -91,7 +91,7 @@ def test_create_unknown_suite_mentions_available_suites(
 
 def test_create_wraps_import_error_and_chains_cause(monkeypatch):
     monkeypatch.setattr(
-        compat, "_env_module_map", {"dummy": ("dummy_mod", "DummyWrapper")}
+        adapters, "_env_module_map", {"dummy": ("dummy_mod", "DummyWrapper")}
     )
 
     def fake_import_module(name: str):
@@ -164,7 +164,7 @@ def test_create_imports_only_the_requested_suite(monkeypatch):
     module_b = types.SimpleNamespace(WrapperB=WrapperB)
 
     monkeypatch.setattr(
-        compat,
+        adapters,
         "_env_module_map",
         {"a": ("a_mod", "WrapperA"), "b": ("b_mod", "WrapperB")},
     )

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_create_integration.py

@@ -1,22 +1,22 @@
-"""Integration tests for envelope.compat.create().
+"""Integration tests for envelope.adapters.create().
 
-These require optional compatibility dependencies (brax/gymnax/navix). They are
-kept separate from the unit tests so a minimal install can still run the suite.
+These require optional adapter dependencies (brax/gymnax/navix). They are kept separate
+from the unit tests so a minimal install can still run the suite.
 """
 
 import pytest
 
-from envelope.compat import create
+from envelope.adapters import create
 from envelope.environment import Environment
 from envelope.wrappers.truncation_wrapper import TruncationWrapper
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 
 def test_create_brax_smoke(prng_key):
     pytest.importorskip("brax")
 
-    from envelope.compat.brax_envelope import BraxEnvelope
+    from envelope.adapters.brax_envelope import BraxEnvelope
 
     env = create("brax::fast")
     assert isinstance(env, TruncationWrapper)
@@ -31,7 +31,7 @@ def test_create_brax_smoke(prng_key):
 def test_create_gymnax_smoke(prng_key):
     pytest.importorskip("gymnax")
 
-    from envelope.compat.gymnax_envelope import GymnaxEnvelope
+    from envelope.adapters.gymnax_envelope import GymnaxEnvelope
 
     env = create("gymnax::CartPole-v1")
     assert isinstance(env, TruncationWrapper)
@@ -46,7 +46,7 @@ def test_create_gymnax_smoke(prng_key):
 def test_create_navix_smoke(prng_key):
     pytest.importorskip("navix")
 
-    from envelope.compat.navix_envelope import NavixEnvelope
+    from envelope.adapters.navix_envelope import NavixEnvelope
 
     env = create("navix::Navix-Empty-5x5-v0")
     assert isinstance(env, TruncationWrapper)
@@ -61,7 +61,7 @@ def test_create_navix_smoke(prng_key):
 def test_create_jumanji_smoke(prng_key):
     pytest.importorskip("jumanji")
 
-    from envelope.compat.jumanji_envelope import JumanjiEnvelope
+    from envelope.adapters.jumanji_envelope import JumanjiEnvelope
 
     env = create("jumanji::Snake-v1")
     assert isinstance(env, TruncationWrapper)
@@ -76,7 +76,7 @@ def test_create_jumanji_smoke(prng_key):
 def test_create_craftax_smoke(prng_key):
     pytest.importorskip("craftax")
 
-    from envelope.compat.craftax_envelope import CraftaxEnvelope
+    from envelope.adapters.craftax_envelope import CraftaxEnvelope
 
     env = create("craftax::Craftax-Symbolic-v1")
     assert isinstance(env, TruncationWrapper)
@@ -91,7 +91,7 @@ def test_create_craftax_smoke(prng_key):
 def test_create_mujoco_playground_smoke(prng_key):
     pytest.importorskip("mujoco_playground")
 
-    from envelope.compat.mujoco_playground_envelope import MujocoPlaygroundEnvelope
+    from envelope.adapters.mujoco_playground_envelope import MujocoPlaygroundEnvelope
 
     env = create("mujoco_playground::CartpoleBalance")
     assert isinstance(env, TruncationWrapper)
@@ -106,7 +106,7 @@ def test_create_mujoco_playground_smoke(prng_key):
 def test_create_kinetix_smoke(prng_key):
     pytest.importorskip("kinetix")
 
-    from envelope.compat.kinetix_envelope import KinetixEnvelope
+    from envelope.adapters.kinetix_envelope import KinetixEnvelope
 
     env = create("kinetix::random")
     assert isinstance(env, TruncationWrapper)

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_gymnax_compat.py

@@ -6,13 +6,13 @@ import jax
 import jax.numpy as jnp
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 pytest.importorskip("gymnax")
 
+from envelope.compat.gymnax_envelope import GymnaxEnvelope, _convert_space
 from gymnax.environments import spaces as gymnax_spaces
 
-from envelope.compat.gymnax_envelope import GymnaxEnvelope, _convert_space
 from envelope.spaces import Continuous, Discrete, PyTreeSpace
 from tests.compat.contract import (
     assert_jitted_rollout_contract,

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_jumanji_compat.py

@@ -1,4 +1,4 @@
-"""Tests for envelope.compat.jumanji_envelope module."""
+"""Tests for envelope.adapters.jumanji_envelope module."""
 
 # ruff: noqa: E402
 
@@ -11,19 +11,19 @@ import jax.numpy as jnp
 import numpy as np
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 pytest.importorskip("jumanji")
 
 from jumanji import specs
 
-import envelope.compat.jumanji_envelope as jumanji_envelope
-from envelope.compat.jumanji_envelope import (
+import envelope.adapters.jumanji_envelope as jumanji_envelope
+from envelope.adapters.jumanji_envelope import (
     JumanjiEnvelope,
     convert_jumanji_spec_to_envelope_space,
 )
 from envelope.spaces import Continuous, Discrete, PyTreeSpace
-from tests.compat.contract import (
+from tests.adapters.contract import (
     assert_jitted_rollout_contract,
     assert_reset_step_contract,
 )

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_kinetix_compat.py

@@ -1,4 +1,4 @@
-"""Tests for envelope.compat.kinetix_envelope module."""
+"""Tests for envelope.adapters.kinetix_envelope module."""
 
 # ruff: noqa: E402
 
@@ -10,17 +10,17 @@ import jax
 import jax.numpy as jnp
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 pytest.importorskip("kinetix")
 
 import kinetix
 from kinetix.environment import EnvParams, StaticEnvParams
 
-from envelope.compat.kinetix_envelope import KinetixEnvelope, _normalize_level_id
+from envelope.adapters.kinetix_envelope import KinetixEnvelope, _normalize_level_id
 from envelope.environment import Info
 from envelope.spaces import Continuous
-from tests.compat.contract import (
+from tests.adapters.contract import (
     assert_jitted_rollout_contract,
     assert_reset_step_contract,
 )
@@ -189,7 +189,7 @@ def test_from_name_rejects_unknown_env_kwargs():
 
 def test_from_name_allows_premade_state_none(monkeypatch: pytest.MonkeyPatch):
     """Current implementation does not guard against missing premade state."""
-    from envelope.compat import kinetix_envelope
+    from envelope.adapters import kinetix_envelope
 
     def mock_load(_level_id: str):
         return None, StaticEnvParams(), EnvParams()
@@ -210,7 +210,7 @@ def test_create_premade_replace_failure_raises(monkeypatch: pytest.MonkeyPatch):
 
     monkeypatch.setattr(EnvParams, "replace", failing_replace)
     monkeypatch.setattr(
-        "envelope.compat.kinetix_envelope.load_from_json_file",
+        "envelope.adapters.kinetix_envelope.load_from_json_file",
         lambda _level_id: (object(), StaticEnvParams(), ep),
     )
 
@@ -218,6 +218,5 @@ def test_create_premade_replace_failure_raises(monkeypatch: pytest.MonkeyPatch):
         KinetixEnvelope.from_name("s/h4_thrust_aim")
 
 
-#
-# NOTE: Level-id normalization tests live in tests/compat/test_kinetix_level_id.py
+# NOTE: Level-id normalization tests live in tests/adapters/test_kinetix_level_id.py
 # to keep this module focused on runtime wrapper behavior.

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_mujoco_playground_compat.py

@@ -1,4 +1,4 @@
-"""Tests for envelope.compat.mujoco_playground_envelope module."""
+"""Tests for envelope.adapters.mujoco_playground_envelope module."""
 
 # ruff: noqa: E402
 
@@ -6,14 +6,14 @@ import jax
 import jax.numpy as jnp
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 pytest.importorskip("mujoco_playground")
 
-from envelope.compat.mujoco_playground_envelope import MujocoPlaygroundEnvelope
+from envelope.adapters.mujoco_playground_envelope import MujocoPlaygroundEnvelope
 from envelope.environment import Info
 from envelope.spaces import Continuous, PyTreeSpace
-from tests.compat.contract import (
+from tests.adapters.contract import (
     assert_jitted_rollout_contract,
     assert_reset_step_contract,
 )

{jax_envelope-0.3.0/tests/compat → jax_envelope-0.4.0/tests/adapters}/test_navix_compat.py

@@ -1,4 +1,4 @@
-"""Tests for envelope.compat.navix_envelope module."""
+"""Tests for envelope.adapters.navix_envelope module."""
 
 # ruff: noqa: E402
 
@@ -6,15 +6,15 @@ import jax
 import jax.numpy as jnp
 import pytest
 
-pytestmark = pytest.mark.compat
+pytestmark = pytest.mark.adapters
 
 pytest.importorskip("navix")
 
 import navix
 
-from envelope.compat.navix_envelope import NavixEnvelope
+from envelope.adapters.navix_envelope import NavixEnvelope
 from envelope.spaces import Continuous, Discrete
-from tests.compat.contract import (
+from tests.adapters.contract import (
     assert_jitted_rollout_contract,
     assert_reset_step_contract,
 )
@@ -160,7 +160,7 @@ def test_unsupported_space_type():
     import jax.numpy as jnp
     from navix import spaces as navix_spaces
 
-    from envelope.compat.navix_envelope import convert_navix_to_envelope_space
+    from envelope.adapters.navix_envelope import convert_navix_to_envelope_space
 
     # Create a mock space that's neither Discrete nor Continuous
     class MockSpace(navix_spaces.Space):
@@ -181,7 +181,7 @@ def test_step_type_conversion(navix_env, prng_key):
     """Test all navix StepType values are correctly converted."""
     import navix
 
-    from envelope.compat.navix_envelope import convert_navix_to_envelope_info
+    from envelope.adapters.navix_envelope import convert_navix_to_envelope_info
 
     env = navix_env
     key = prng_key
@@ -224,7 +224,7 @@ def test_discrete_space_conversion():
     """Test conversion of discrete spaces from navix to envelope."""
     from navix import spaces as navix_spaces
 
-    from envelope.compat.navix_envelope import convert_navix_to_envelope_space
+    from envelope.adapters.navix_envelope import convert_navix_to_envelope_space
 
     # Create a navix Discrete space
     navix_discrete = navix_spaces.Discrete.create(10, shape=(3,), dtype=jnp.int32)
@@ -245,7 +245,7 @@ def test_continuous_space_conversion():
     """Test conversion of continuous spaces from navix to envelope."""
     from navix import spaces as navix_spaces
 
-    from envelope.compat.navix_envelope import convert_navix_to_envelope_space
+    from envelope.adapters.navix_envelope import convert_navix_to_envelope_space
 
     # Create a navix Continuous space
     navix_continuous = navix_spaces.Continuous.create(

{jax_envelope-0.3.0 → jax_envelope-0.4.0}/uv.lock

@@ -1056,14 +1056,14 @@ wheels = [
 
 [[package]]
 name = "jax-envelope"
-version = "0.3.0"
+version = "0.4.0"
 source = { editable = "." }
 dependencies = [
     { name = "jax" },
 ]
 
 [package.dev-dependencies]
-compat = [
+adapters = [
     { name = "brax" },
     { name = "craftax" },
     { name = "gymnax" },
@@ -1083,7 +1083,7 @@ dev = [
 requires-dist = [{ name = "jax", specifier = ">=0.5.0" }]
 
 [package.metadata.requires-dev]
-compat = [
+adapters = [
     { name = "brax", specifier = ">=0.13.0" },
     { name = "craftax", specifier = ">=1.4.3" },
     { name = "gymnax", git = "https://github.com/RobertTLange/gymnax.git" },