Skip to content

Up n Down#210

Open
BastiJilge wants to merge 76 commits into
k4ntz:devfrom
BastiJilge:feature/UpNDown-implementation
Open

Up n Down#210
BastiJilge wants to merge 76 commits into
k4ntz:devfrom
BastiJilge:feature/UpNDown-implementation

Conversation

@BastiJilge

Copy link
Copy Markdown

python scripts/play.py -g UpNDown

Known Bugs:

  • The player teleports onto the nearest road instead of gradually getting closer to it after a jump
  • The path the cars take can sometimes appear to go off road

@github-actions

github-actions Bot commented Dec 21, 2025

Copy link
Copy Markdown
📁 Previous CI results (run #20799774671)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/20414284420

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2025-12-21 19:00:37 UTC.

@Dan041k Dan041k self-assigned this Jan 6, 2026
@aswiso

aswiso commented Jan 6, 2026

Copy link
Copy Markdown

ALE Similarity: 2.5 / 4

  • (-1) Only the first round is fully playable. After collecting all flags, the player is awarded the maximum score (999999), but the game continues instead of properly resetting or progressing to a new round.
  • (-0.5) Several smaller inconsistencies compared to the original game:
    • Some animations are missing, such as destruction effects and the player’s car visibly moving from right to left.
    • Jumping in the original game feels more responsive and has less delay. Additionally, the original version provides a short window after landing with bigger collision force than in the new version, allowing the player to destroy enemy cars.
    • Enemy car behavior is less challenging than in the original game. Sudden backward movements and random accelerations occur more frequently in the original, and there appears to be a correlation between the player moving backward and enemy cars reacting accordingly.
    • In the original game, gaining 10 000 points awards the player an extra life, which is not currently implemented.
    • Acceleration and deceleration mechanics require further tuning to better match the original gameplay. For example, crossing the dark grey “bridge” is possible with enough speed in ALE but requires a jump in the current implementation.

Implementation Quality: 4 / 4

The code is fully JAX-native, with no Python loops used in the core logic and all state updates expressed through pure functions. The overall structure is clear and well modularized, and the code is thoroughly commented, making peer review and further development straightforward. The game loads instantly and runs smoothly, with no noticeable performance issues. Additionally, the current implementation is sufficiently modular and flexible to support gameplay modifications.

Overall, this is already a solid and well-structured reimplementation of the original UpNDown. While some gameplay details differ a little bit from the original, the core systems are there and behave accordingly. Thanks to the clean, modular, and JAX-compatible design, the implementation is well suited for further balancing and extension.

@jakobsteinke

jakobsteinke commented Jan 6, 2026

Copy link
Copy Markdown

Hi, this is the review of the team working on Darkchambers:

ALE-Similarity: 3 / 4

Overall already great. Here are some points we noticed:

Gameplay & Mechanics

  • Acceleration behavior (Up input)
    In the original game, holding Up behaves like acceleration: the car speeds up while the button is pressed and gradually slows down once released.
    In the current implementation, speed appears to remain constant after a single button press (vice versa for Down input).

  • Jump input handling
    In the original game, holding the jump button (Space) only triggers a single jump. Continuous jumping while holding the button is not possible.

  • Jump cooldown dynamics
    The jump cooldown in the original game appears to depend on the current speed of the car, whereas in the current version it seems to be fixed or speed-independent.

  • Limited to one level
    Once you have all flags, the game should reset and start a new round.

  • Changing direction
    When moving up, pressing down in the original game does not immediately stop or reverse the car’s movement. Instead, deceleration is gradual and depends on how long the button is held (vice versa for changing from down to up input).

  • Jump collision tolerance
    When jumping onto other cars, collision detection could be slightly more forgiving. In the original game, it is easier to land hits on other cars, which makes jumps feel more reliable and less punishing.

  • Extra life
    Getting to 10 000 points should give the player an extra life.

  • Animations (optional polish)
    If time permits, adding more animations would improve visual feedback and overall game feel.


Code Quality: 3.5 / 4

What’s good

  • State structure is clean and JAX-friendly
  • Collision detection is properly vectorized
  • Road constants are precomputed

Where you could optimize

  1. Flag collision check uses vmap for 8 flags (Line 679)

    • Overkill for small fixed-size arrays
    • Could be replaced with vectorized ops, e.g.:
      y_dist = jnp.abs(state.flags.y - player_y)
      collisions = (y_dist < threshold) & (x_dist < threshold) & ...
  2. Road geometry recalculated repeatedly (Lines 336–348 and others)

    • Computing X position on road happens in _flag_step, _collectible_step, _enemy_step
    • Caching slope/intercept when the road segment changes would likely be better
  3. Type conversions in render loops (Line 2060+)

    • enemy_x.astype(jnp.int32) called repeatedly
    • Could be converted once before the loop

Performance: 4 / 4

  • Environment runs smoothly and responsively during gameplay
  • No freezes, crashes, or noticeable slowdowns observed (aside from known dev bugs)
  • Player input is handled reliably
  • Loading times are short
  • Sprites load correctly and the game works as expected

@github-actions

github-actions Bot commented Jan 7, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #21547230802)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/20799774671

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ❌

The PR changes files that should not be changed:

  • renovate.json
  • scripts/spriteEditor/convert_icon.py
  • scripts/spriteEditor/spriteEditor.py
  • src/jaxatari/rendering/jax_rendering_utils.py

Please ensure that only allowed files are modified. Any changes in the src/jaxatari/games/ directory are allowed.

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-01-07 23:30:55 UTC.

@github-actions

github-actions Bot commented Jan 31, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #22721145414)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/21547230802

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ❌

The PR changes files that should not be changed:

  • renovate.json
  • scripts/spriteEditor/convert_icon.py
  • scripts/spriteEditor/spriteEditor.py
  • src/jaxatari/rendering/jax_rendering_utils.py

Please ensure that only allowed files are modified. Any changes in the src/jaxatari/games/ directory are allowed.

Framework Tests ❌

Some framework tests failed. Please check the details below:

freeway ❌
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: github-actions-annotate-failures-0.3.0, sugar-1.1.1, syrupy-4.9.1, xdist-3.8.0, jaxtyping-0.3.7
created: 2/2 workers
2 workers [157 items]

.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
..::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=505::test_native_downscaling_hot_swap[freeway]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: FreewayRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=567::test_native_downscaling_grayscale[freeway]%0A%0ATypeError: FreewayRenderer.__init__() got an unexpected keyword argument 'config'
F.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
....::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/modification.py,line=746::Modding System: Using legacy '_replace()' for 'freeway' constants. Please migrate constants to 'flax.struct.PyTreeNode' (and the state to flax.struct.dataclass/PyTreeNode) for better performance and future compatibility.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
. [ 45%]
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
sssssssssssssss.::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_freeway.py,line=196::Performance Warning: JaxFreeway.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=/home/runner/work/JAXAtari/JAXAtari/sr...*[Comment body truncated]*

</details>

@BastiJilge

Copy link
Copy Markdown
Author

Planed Mods:

Simple mods:

  1. Allow jumping while moving backwards
  2. Remove step roads
  3. Allow higher player speed
  4. Spawn more collectibles
  5. Add a minimum gap in between car spawns

Difficult Mods:

  1. Cars only spawn on the right side of the screen
  2. Increase Car spawn as time progresses in a level
  3. Collectibles are worth less the more time is spend on a level

If there is a problem with some of the planed mods please let us know @PaulSeitz

@github-actions

github-actions Bot commented Mar 5, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #22763292280)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/22721145414

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ❌

Some framework tests failed. Please check the details below:

upndown ❌
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [143 items]

sssssss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.....::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
. [ 50%]
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ssssssssssssssss.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warnin...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 6, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #22763789740)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/22763292280

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ❌

Some framework tests failed. Please check the details below:

upndown ❌
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [143 items]

sssssss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.....::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=183::test_multi_reward_log_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=183::test_multi_reward_log_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=254::test_flatten_observation_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=254::test_flatten_observation_wrapper[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=314::test_log_wrapper_with_flatten_observation[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=314::test_log_wrapper_with_flatten_observation[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=345::test_flatten_observation_wrapper_space_structure[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=345::test_flatten_observation_wrapper_space_structure[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=433::test_atari_wrapper_features_and_pixel_preprocessing[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=433::test_atari_wrapper_features_and_pixel_preprocessing[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=181::TestBasicAPI.test_determinism[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=181::TestBasicAPI.test_determinism[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-upndown]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1870%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_upndown.py:1871%0A%0AThe output of true_fun at path .player_car.speed has type float32[] but the corresponding output of false_fun has type int32[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 6, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23384723074)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/22763789740

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ❌

Some framework tests failed. Please check the details below:

upndown ❌
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [143 items]

sssssss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.....::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::error file=tests/test_core_and_wrappers.py,line=183::test_multi_reward_log_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=183::test_multi_reward_log_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=254::test_flatten_observation_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=254::test_flatten_observation_wrapper[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=314::test_log_wrapper_with_flatten_observation[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=314::test_log_wrapper_with_flatten_observation[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=345::test_flatten_observation_wrapper_space_structure[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=345::test_flatten_observation_wrapper_space_structure[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=433::test_atari_wrapper_features_and_pixel_preprocessing[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=433::test_atari_wrapper_features_and_pixel_preprocessing[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=181::TestBasicAPI.test_determinism[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=181::TestBasicAPI.test_determinism[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Atari-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Atari-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-upndown]%0A%0ANameError: name 'can_accelerate' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 21, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23385404396)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23384723074

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ❌

The PR changes files that should not be changed:

  • .github/workflows/reusable-run-tests.yml
  • games_covered.md
  • pyproject.toml
  • renovate.json
  • scripts/benchmarks/agent_performance_comparison.py
  • scripts/benchmarks/config/alg/ppo_jaxatari_object.yaml
  • scripts/benchmarks/config/alg/pqn_jaxatari_object.yaml
  • scripts/benchmarks/config/alg/pqn_jaxatari_pixel.yaml
  • scripts/benchmarks/config/config.yaml
  • scripts/benchmarks/ppo_agent.py
  • scripts/benchmarks/ppo_agent_old.py
  • scripts/benchmarks/ppo_test.py
  • scripts/benchmarks/pqn_agent.py
  • scripts/benchmarks/pqn_test.py
  • scripts/benchmarks/pure_performance_comparison.py
  • scripts/benchmarks/train_jaxatari_agent.py
  • scripts/benchmarks/train_ocatari_agent.py
  • scripts/benchmarks/train_utils.py
  • scripts/gameplay_comparison.py
  • scripts/play.py
  • scripts/spriteEditor/convert_icon.py
  • scripts/spriteEditor/spriteEditor.py
  • scripts/utils.py
  • src/jaxatari/__init__.py
  • src/jaxatari/core.py
  • src/jaxatari/modification.py
  • src/jaxatari/rendering/jax_rendering_utils.py
  • src/jaxatari/rendering/jax_rendering_utils_legacy.py
  • src/jaxatari/wrappers.py
  • tests/__snapshots__/test_regression.ambr
  • tests/conftest.py
  • tests/test_spaces.py
  • train_agents.md
  • uv.lock

Please ensure that only allowed files are modified. Any changes in the src/jaxatari/games/ directory are allowed.

Framework Tests ❌

Some framework tests failed. Please check the details below:

tennis ✅
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [148 items]

.......................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
................................................. [ 48%]
............ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
....s..::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
...::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
. [ 97%]
....                                                                     [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tennis]
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py:317: UserWarning: WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
    logger.warn(

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tennis]
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py:434: UserWarning: WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
    logger.warn(

tests/test_spaces.py::test_discrete_space
tests/test_spaces.py::test_box_space
tests/test_spaces.py::test_dict_space
tests/test_spaces.py::test_tuple_space
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:82: UserWarning: Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
    super().__init__(consts)

tests/test_spaces.py::test_discrete_space
tests/test_spaces.py::test_box_space
tests/test_spaces.py::test_dict_space
tests/test_spaces.py::test_tuple_space
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/core.py:182: DeprecationWarning: Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
    _warn_deprecated_obs_to_flat_array(env)

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
=========================== short test summary info ============================
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 131 passed, 17 skipped, 11 warnings in 163.87s (0:02:43) ===========

frostbite ✅
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
..::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=82::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.....s.. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[frostbite]
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py:317: UserWarning: WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
    logger.warn(

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[frostbite]
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py:434: UserWarning: WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
    logger.warn(

tests/test_spaces.py::test_discrete_space
tests/test_spaces.py::test_box_space
tests/test_spaces.py::test_dict_space
tests/test_spaces.py::test_tuple_space
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:82: UserWarning: Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
    super().__init__(consts)

tests/test_spaces.py::test_discrete_space
tests/test_spaces.py::test_box_space
tests/test_spaces.py::test_dict_space
tests/test_spaces.py::test_tuple_space
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/core.py:182: DeprecationWarning: Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
    _warn_deprecated_obs_to_flat_array(env)

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
=========================== short test summary info ============================
SKIPPED [1] tests/test_all_mods.py:188: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:234: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:309: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:333: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:350: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:396: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:676: Game frostbite not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 11 warnings in 592.55s (0:09:52) ===========

asteroids ❌
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s::error file=tests/test_all_mods.py,line=694::TestDatatypeConsistency.test_constants_are_pytree_node[asteroids]%0A%0AAssertionError: Constants should be flax.struct.PyTreeNode, got <class 'jax_asteroids.AsteroidsConstants'>. Please refactor from NamedTuple to PyTreeNode.%0Aassert False
::error file=tests/test_all_mods.py,line=694::TestDatatypeConsistency.test_constants_are_pytree_node[asteroids]%0A%0AAssertionError: Constants should be flax.struct.PyTreeNode, got <class 'jax_asteroids.AsteroidsConstants'>. Please refactor from NamedTuple to PyTreeNode.%0Aassert False
F::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=712::TestDatatypeConsistency.test_state_is_struct_dataclass[asteroids]%0A%0AAssertionError: State should be @struct.dataclass (PyTreeNode), got <class 'jax_asteroids.AsteroidsState'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=712::TestDatatypeConsistency.test_state_is_struct_dataclass[asteroids]%0A%0AAssertionError: State should be @struct.dataclass (PyTreeNode), got <class 'jax_asteroids.AsteroidsState'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=732::TestDatatypeConsistency.test_observation_is_struct_dataclass[asteroids]%0A%0AAssertionError: Observation should be @struct.dataclass (PyTreeNode), got <class 'jax_asteroids.AsteroidsObservation'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=732::TestDatatypeConsistency.test_observation_is_struct_dataclass[asteroids]%0A%0AAssertionError: Observation should be @struct.dataclass (PyTreeNode), got <class 'jax_asteroids.AsteroidsObservation'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=752::TestDatatypeConsistency.test_info_is_struct_dataclass[asteroids]%0A%0AAssertionError: Info should be @struct.dataclass (PyTreeNode), got <class 'jax_asteroids.AsteroidsInfo'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=752::TestDatatypeConsistency.test_info_is_struct_dataclass[asteroids]%0A%0AAssertionError: Info should be @struct.dataclass (PyTreeNode), got <class 'jax_asteroids.AsteroidsInfo'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[asteroids]%0A%0AAssertionError: Observation type: <class 'jax_asteroids.AsteroidsObservation'>%0Aassert False
::error file=tests/test_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[asteroids]%0A%0AAssertionError: Observation type: <class 'jax_asteroids.AsteroidsObservation'>%0Aassert False
F::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.....s::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[asteroids]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: AsteroidsRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[asteroids]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: AsteroidsRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[asteroids]%0A%0ATypeError: AsteroidsRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[asteroids]%0A%0ATypeError: AsteroidsRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_asteroids.py,line=210::Performance Warning: JaxAsteroids.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.d...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 21, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23387019406)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23385404396

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ❌

The PR changes files that should not be changed:

  • .github/workflows/reusable-run-tests.yml
  • pyproject.toml
  • renovate.json
  • scripts/benchmarks/agent_performance_comparison.py
  • scripts/benchmarks/config/alg/ppo_jaxatari_object.yaml
  • scripts/benchmarks/config/alg/pqn_jaxatari_object.yaml
  • scripts/benchmarks/ppo_agent.py
  • scripts/benchmarks/ppo_agent_old.py
  • scripts/benchmarks/pure_performance_comparison.py
  • scripts/benchmarks/train_jaxatari_agent.py
  • scripts/benchmarks/train_ocatari_agent.py
  • scripts/benchmarks/train_utils.py
  • scripts/play.py
  • scripts/spriteEditor/convert_icon.py
  • scripts/spriteEditor/spriteEditor.py
  • scripts/utils.py
  • src/jaxatari/core.py
  • src/jaxatari/rendering/jax_rendering_utils.py
  • src/jaxatari/wrappers.py
  • tests/__snapshots__/test_regression.ambr
  • tests/conftest.py
  • tests/test_spaces.py
  • train_agents.md
  • uv.lock

Please ensure that only allowed files are modified. Any changes in the src/jaxatari/games/ directory are allowed.

Framework Tests ❌

Some framework tests failed. Please check the details below:

asteroids ❌
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

ssssEsEsEEEEEEEEEEEEEEEEEEEEEE.E::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
sEEEEEEEEEEE.E.E..E.EEEEEEEEEEEEEEEEEEEE [ 49%]
EEEEEEEEEEEEEEssssssssssssEEEEEEEEEssssEEEEEEEEEEEEEEEEEEEEEEs::error file=tests/test_spaces.py,line=17::test_discrete_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=17::test_discrete_space%0A%0ANameError: name 'asset_config' is not defined
F::error file=tests/test_spaces.py,line=83::test_dict_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=83::test_dict_space%0A%0ANameError: name 'asset_config' is not defined
F::error file=tests/test_spaces.py,line=49::test_box_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=49::test_box_space%0A%0ANameError: name 'asset_config' is not defined
F.::error file=tests/test_spaces.py,line=116::test_tuple_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=116::test_tuple_space%0A%0ANameError: name 'asset_config' is not defined
F..... [ 99%]
.                                                                        [100%]
==================================== ERRORS ====================================
_ ERROR at setup of TestDatatypeConsistency.test_constants_are_pytree_node[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_state_is_struct_dataclass[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_observation_is_struct_dataclass[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_info_is_struct_dataclass[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_environment_no_verbose_parameter[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_environment_methods_no_verbose_parameter[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_datatype_consistency_across_operations[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of test_pixel_obs_wrapper_with_stacked_frames[asteroids] ____
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
______ ERROR at setup of test_pixel_and_object_centric_wrapper[asteroids] ______
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___________ ERROR at setup of test_object_centric_wrapper[asteroids] ___________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________________ ERROR at setup of test_log_wrapper[asteroids] _________________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__________ ERROR at setup of test_multi_reward_log_wrapper[asteroids] __________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________ ERROR at setup of test_flatten_observation_wrapper[asteroids] _________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
____ ERROR at setup of test_log_wrapper_with_flatten_observation[asteroids] ____
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of test_flatten_observation_wrapper_space_structure[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of test_atari_wrapper_features_and_pixel_preprocessing[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________ ERROR at setup of test_native_downscaling_hot_swap[asteroids] _________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________ ERROR at setup of test_native_downscaling_grayscale[asteroids] ________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of TestModifications.test_constants_structure[asteroids] ____
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestModifications.test_constants_can_be_overridden_conceptually[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestModifications.test_mod_system_loads_without_error[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestModifications.test_environment_works_after_mods[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__________ ERROR at setup of test_specific_game_mods_load[asteroids] ___________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
____________ ERROR at setup of TestBasicAPI.test_spaces[asteroids] _____________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_____________ ERROR at setup of TestBasicAPI.test_reset[asteroids] _____________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_____________ ERROR at setup of TestBasicAPI.test_step[asteroids] ______________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__________ ERROR at setup of TestBasicAPI.test_determinism[asteroids] __________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
____________ ERROR at setup of TestBasicAPI.test_render[asteroids] _____________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
______ ERROR at setup of TestBasicAPI.test_episode_completion[asteroids] _______
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestJaxTransforms.test_jit_compilation[Atari-asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestJaxTransforms.test_jit_compilation[Pixel-asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[Atari-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_log_wrapper_tracking[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_multi_reward_log_wrapper[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_atari_wrapper_features[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_consistency[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_log_wrapper_edge_cases[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_observation_consistency[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_invalid_action_handling[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_extreme_reward_values[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[PixelAndObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymnasiumApiCompliance.test_gymnasium_env_checker[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestGymnasiumApiCompliance.test_reset_method[asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[NormalizedPixel-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of TestGymnasiumApiCompliance.test_step_method[asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[LoggedFlattenedPixelAndObject-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestGymnasiumApiCompliance.test_render_method[asteroids] __
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[MultiRewardLogged-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymnasiumApiCompliance.test_seeding_and_determinism[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymWrapperIntegration.test_frame_stack_wrapper[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymWrapperIntegration.test_time_limit_wrapper[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of TestJaxTransforms.test_jittable_transition[asteroids] ____
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymWrapperIntegration.test_preprocessing_wrappers[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
=========================== short test summary info ============================
SKIPPED [1] tests/test_all_mods.py:188: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:234: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:309: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:333: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:350: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:396: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_constants_are_pytree_node[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_state_is_struct_dataclass[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_observation_is_struct_dataclass[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_info_is_struct_dataclass[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_environment_no_verbose_parameter[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_environment_methods_no_verbose_parameter[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_datatype_consistency_across_operations[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_pixel_obs_wrapper_with_stacked_frames[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_pixel_and_object_centric_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_object_centric_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_log_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_multi_reward_log_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_flatten_observation_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_log_wrapper_with_flatten_observation[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_flatten_observation_wrapper_space_structure[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_atari_wrapper_features_and_pixel_preprocessing[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_native_downscaling_hot_swap[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_native_downscaling_grayscale[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_constants_structure[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_constants_can_be_overridden_conceptually[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_mod_system_loads_without_error[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_environment_works_after_mods[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::test_specific_game_mods_load[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_spaces[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_reset[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_step[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_determinism[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_render[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_episode_completion[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_flatten_observation_wrapper_space_structure[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_normalize_observation_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_tracking[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_multi_reward_log_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_atari_wrapper_features[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_consistency[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_edge_cases[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_observation_consistency[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_invalid_action_handling[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_extreme_reward_values[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_reset_method[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_render_method[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_frame_stack_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestJaxTransforms::test_jittable_transition[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_preprocessing_wrappers[asteroids] - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_discrete_space - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_dict_space - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_box_space - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_tuple_space - NameError: name 'asset_config' is not defined
======= 4 failed, 13 passed, 24 skipped, 1 warning, 104 errors in 10.83s =======

upndown ❌
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.....s::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=188::Performance Warning: JaxUpNDown.consts is a '...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 21, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23387247904)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23387019406

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ❌

The PR changes files that should not be changed:

  • .github/workflows/reusable-run-tests.yml
  • pyproject.toml
  • renovate.json
  • scripts/benchmarks/agent_performance_comparison.py
  • scripts/benchmarks/config/alg/ppo_jaxatari_object.yaml
  • scripts/benchmarks/config/alg/pqn_jaxatari_object.yaml
  • scripts/benchmarks/ppo_agent.py
  • scripts/benchmarks/ppo_agent_old.py
  • scripts/benchmarks/pure_performance_comparison.py
  • scripts/benchmarks/train_jaxatari_agent.py
  • scripts/benchmarks/train_ocatari_agent.py
  • scripts/benchmarks/train_utils.py
  • scripts/play.py
  • scripts/spriteEditor/convert_icon.py
  • scripts/spriteEditor/spriteEditor.py
  • scripts/utils.py
  • src/jaxatari/core.py
  • src/jaxatari/rendering/jax_rendering_utils.py
  • src/jaxatari/wrappers.py
  • tests/__snapshots__/test_regression.ambr
  • tests/conftest.py
  • tests/test_spaces.py
  • train_agents.md
  • uv.lock

Please ensure that only allowed files are modified. Any changes in the src/jaxatari/games/ directory are allowed.

Framework Tests ❌

Some framework tests failed. Please check the details below:

asteroids ❌
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

sssssEsEEEEEEEEEEEEEEEEEEEEEEEEE.E::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
sEEEEEEEEEEE.E..E.E.EEEEEEEEEEEEEEEEEE [ 49%]
EEEEEEEEEEEEEEEssssssssssssEEEEEEEssssEEEEEEEEEEEEEEEEEEEEEEEs::error file=tests/test_spaces.py,line=17::test_discrete_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=17::test_discrete_space%0A%0ANameError: name 'asset_config' is not defined
F::error file=tests/test_spaces.py,line=83::test_dict_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=83::test_dict_space%0A%0ANameError: name 'asset_config' is not defined
F::error file=tests/test_spaces.py,line=49::test_box_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=49::test_box_space%0A%0ANameError: name 'asset_config' is not defined
F::error file=tests/test_spaces.py,line=116::test_tuple_space%0A%0ANameError: name 'asset_config' is not defined
::error file=tests/test_spaces.py,line=116::test_tuple_space%0A%0ANameError: name 'asset_config' is not defined
F...... [ 99%]
.                                                                        [100%]
==================================== ERRORS ====================================
_ ERROR at setup of TestDatatypeConsistency.test_constants_are_pytree_node[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_state_is_struct_dataclass[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_observation_is_struct_dataclass[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_info_is_struct_dataclass[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_environment_no_verbose_parameter[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_environment_methods_no_verbose_parameter[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestDatatypeConsistency.test_datatype_consistency_across_operations[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of test_pixel_obs_wrapper_with_stacked_frames[asteroids] ____
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
______ ERROR at setup of test_pixel_and_object_centric_wrapper[asteroids] ______
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___________ ERROR at setup of test_object_centric_wrapper[asteroids] ___________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________________ ERROR at setup of test_log_wrapper[asteroids] _________________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__________ ERROR at setup of test_multi_reward_log_wrapper[asteroids] __________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________ ERROR at setup of test_flatten_observation_wrapper[asteroids] _________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
____ ERROR at setup of test_log_wrapper_with_flatten_observation[asteroids] ____
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of test_flatten_observation_wrapper_space_structure[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of test_atari_wrapper_features_and_pixel_preprocessing[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________ ERROR at setup of test_native_downscaling_hot_swap[asteroids] _________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
________ ERROR at setup of test_native_downscaling_grayscale[asteroids] ________
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of TestModifications.test_constants_structure[asteroids] ____
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestModifications.test_constants_can_be_overridden_conceptually[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestModifications.test_mod_system_loads_without_error[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestModifications.test_environment_works_after_mods[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__________ ERROR at setup of test_specific_game_mods_load[asteroids] ___________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
____________ ERROR at setup of TestBasicAPI.test_spaces[asteroids] _____________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_____________ ERROR at setup of TestBasicAPI.test_reset[asteroids] _____________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_____________ ERROR at setup of TestBasicAPI.test_step[asteroids] ______________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__________ ERROR at setup of TestBasicAPI.test_determinism[asteroids] __________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
____________ ERROR at setup of TestBasicAPI.test_render[asteroids] _____________
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
______ ERROR at setup of TestBasicAPI.test_episode_completion[asteroids] _______
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestJaxTransforms.test_jit_compilation[Atari-asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestJaxTransforms.test_jit_compilation[Pixel-asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[Atari-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[FlattenedObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[NormalizedPixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[LoggedFlattenedPixelAndObject-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_log_wrapper_tracking[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_multi_reward_log_wrapper[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_consistency[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_atari_wrapper_features[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_observation_consistency[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestAdvancedWrapperFeatures.test_log_wrapper_edge_cases[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[Atari-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_invalid_action_handling[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[Pixel-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_extreme_reward_values[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[ObjectCentric-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[PixelAndObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[MultiRewardLogged-asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymnasiumApiCompliance.test_gymnasium_env_checker[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[NormalizedPixel-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestGymnasiumApiCompliance.test_reset_method[asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[LoggedFlattenedPixelAndObject-asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of TestGymnasiumApiCompliance.test_step_method[asteroids] ___
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
__ ERROR at setup of TestGymnasiumApiCompliance.test_render_method[asteroids] __
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymWrapperIntegration.test_preprocessing_wrappers[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymnasiumApiCompliance.test_seeding_and_determinism[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymWrapperIntegration.test_frame_stack_wrapper[asteroids] _
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
_ ERROR at setup of TestGymWrapperIntegration.test_time_limit_wrapper[asteroids] _
[gw0] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
___ ERROR at setup of TestJaxTransforms.test_jittable_transition[asteroids] ____
[gw1] linux -- Python 3.11.15 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NameError: name 'asset_config' is not defined
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_pong.py:589: NameError: name 'asset_config' is not defined
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
=========================== short test summary info ============================
SKIPPED [1] tests/test_all_mods.py:188: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:234: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:309: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:333: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:350: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:396: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_constants_are_pytree_node[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_state_is_struct_dataclass[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_observation_is_struct_dataclass[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_info_is_struct_dataclass[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_environment_no_verbose_parameter[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_environment_methods_no_verbose_parameter[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestDatatypeConsistency::test_datatype_consistency_across_operations[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_pixel_obs_wrapper_with_stacked_frames[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_pixel_and_object_centric_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_object_centric_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_log_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_multi_reward_log_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_flatten_observation_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_log_wrapper_with_flatten_observation[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_flatten_observation_wrapper_space_structure[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_atari_wrapper_features_and_pixel_preprocessing[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_native_downscaling_hot_swap[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_core_and_wrappers.py::test_native_downscaling_grayscale[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_constants_structure[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_constants_can_be_overridden_conceptually[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_mod_system_loads_without_error[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::TestModifications::test_environment_works_after_mods[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_all_mods.py::test_specific_game_mods_load[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_spaces[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_reset[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_step[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_determinism[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_render[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestBasicAPI::test_episode_completion[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_flatten_observation_wrapper_space_structure[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_normalize_observation_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_tracking[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_multi_reward_log_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_consistency[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_atari_wrapper_features[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_observation_consistency[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_edge_cases[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[Atari-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_invalid_action_handling[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[Pixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_extreme_reward_values[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[ObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[PixelAndObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[MultiRewardLogged-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[FlattenedObjectCentric-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[NormalizedPixel-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_reset_method[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[LoggedFlattenedPixelAndObject-asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_render_method[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_preprocessing_wrappers[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_frame_stack_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[asteroids] - NameError: name 'asset_config' is not defined
ERROR tests/test_funcenv_adapter.py::TestJaxTransforms::test_jittable_transition[asteroids] - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_discrete_space - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_dict_space - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_box_space - NameError: name 'asset_config' is not defined
FAILED tests/test_spaces.py::test_tuple_space - NameError: name 'asset_config' is not defined
======= 4 failed, 13 passed, 24 skipped, 1 warning, 104 errors in 10.14s =======

upndown ❌
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.....s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a '...*[Comment body truncated]*

</details>

@BastiJilge BastiJilge force-pushed the feature/UpNDown-implementation branch from fc84b29 to 68443a2 Compare March 21, 2026 19:42
@github-actions

github-actions Bot commented Mar 21, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23387375382)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23387247904

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ❌

Some framework tests failed. Please check the details below:

upndown ❌
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.....::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
Fs::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=320::test_log_wrapper_with_flatten_observation[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=320::test_log_wrapper_with_flatten_observation[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=181::TestBasicAPI.test_determinism[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=181::TestBasicAPI.test_determinism[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Atari-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Atari-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-upndown]%0A%0ANameError: name 'steep_progress' is not defined%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_upndown.py,line=220::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the sta...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 21, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23410964625)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23387375382

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ❌

Some framework tests failed. Please check the details below:

upndown ❌
============================= test session starts ==============================
platform linux -- Python 3.11.15, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ss::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.....s::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[upndown]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[upndown]%0A%0ATypeError: UpNDownRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
. [ 49%]
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
ssssssssssssssss.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_upndown.py,line=216::Performance Warning: JaxUpNDown.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warnin...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 22, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23465833201)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23410964625

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-22 19:51:55 UTC.

@github-actions

github-actions Bot commented Mar 24, 2026

Copy link
Copy Markdown
📁 Previous CI results (run #23687725803)

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23465833201

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-24 00:00:29 UTC.

@github-actions

Copy link
Copy Markdown

Test Report

This comment was generated automatically by a GitHub Action. It summarizes the test results for this pull request. The GitHub Action run can be found here:

https://github.com/k4ntz/JAXAtari/actions/runs/23687725803

Base Branch ✅

The PR's base branch is dev.
The expected base branch is dev.

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-28 15:09:54 UTC.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants