First Submission: Ms. Pac-Man by DimiM99 · Pull Request #218 · k4ntz/JAXAtari

DimiM99 · 2025-12-21T21:59:54Z

🎮 Add Ms. Pac-Man Game to JAXAtari

Summary

Added a new Ms. Pac-Man game implementation to the JAXAtari repository with partially complete game logic, visual rendering, and JAX compatibility.

✅ Features Implemented

🎯 Core Game Mechanics

Full Ms. Pac-Man movement and collision detection
Ghost AI with 2 ghosts (red and pink) (partially complete)
Pellet collection and power pellet system (partially complete)
Lives system with 3 initial lives
Score tracking and game state management

🎨 Visual System

Custom pixel font for text rendering
Start screen with "PRESS START" overlay
Game over screen with "GAME OVER" overlay
Lives display using Ms. Pac-Man sprites (grey when lost)
Classic arcade color scheme: black background, blue corridors, red walls

🔧 Technical Features

Full JAX compatibility with JIT compilation (partially complete)
Proper maze rendering with layered color masks
Sprite-based character rendering
Screen phase management (start → playing → game over)
Input handling for screen transitions

📁 New Files Added

src/jaxatari/games/jax_mspacman.py - Complete Ms. Pac-Man game implementation

🎮 How to Play

add

"mspacman": "jaxatari.games.jax_mspacman",

to GAME_MODULES dict/map in core.py
then run

python scripts/play.py -g mspacman

Start screen: Press any key to begin
Gameplay: Arrow keys or WASD to move
Game over: Press any key to return to start screen

🏗️ Architecture

Built on JAXAtari framework with JAXEnvironment
Uses NamedTuple for state management
Implements proper observation/action spaces
Compatible with existing training pipelines

👨‍💻👨‍💻 Team

Dmitri Moscoglo
Benedikt Wagenlehner

…ript TODOgit add .! Tests and visuals

- sprites for mx pac man ghosts - pallets

github-actions · 2025-12-21T22:07:24Z

📁 Previous CI results (run #20416468380)

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/20416400274

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:

src/jaxatari/core.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 2025-12-21 22:07:23 UTC.

github-actions · 2025-12-21T22:12:37Z

📁 Previous CI results (run #20416627938)

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/20416468380

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:

src/jaxatari/core.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 2025-12-21 22:12:35 UTC.

github-actions · 2025-12-21T22:25:49Z

📁 Previous CI results (run #20416639198)

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/20416627938

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:

src/jaxatari/core.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 2025-12-21 22:25:47 UTC.

github-actions · 2025-12-21T22:26:52Z

📁 Previous CI results (run #20416692689)

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/20416639198

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 22:26:51 UTC.

github-actions · 2025-12-21T22:31:31Z

📁 Previous CI results (run #20826471586)

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/20416692689

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 22:31:28 UTC.

gixerl · 2025-12-29T20:18:43Z

Hey,
First of all, your current implementation provides a decent foundation :)
However, it still differs from ALE in several important ways. Here is a side-by-side comparison:

Screencast_From_2025-12-29_17-08-57.mp4

ALE-similarity: 1.5/4

Strengths:

The level 1 map layout is implemented correctly.
The Sprites for Ms. Pacman and the ghosts are correct
HP gets reduced correctly when dying.
The red dots are collectable and positioned in the center of each path.

Differences:

Your game currently resets after 2000 steps(implemented by the max_steps constant), making it practically impossible to finish. In the ALE this is not the case.
In the ALE, there are further levels when beating level 1, some even having different map layouts. In your implementation, the game just ends after beating level 1.
a lot of sprites in the environment differ significantly from the ALE. You should extract all the sprites from the ALE with the frame extractor tool.
The UI looks different and is at a different position
There is no "PRESS START" text in the ALE
Player speed is currently too high compared to the ALE
Input buffering is missing: the agent turns and stops in place rather than continuing to the next available intersection, making navigation difficult
In the ALE the number of hearts displayed act as the lives you have in addition to your current life. As soon as you die, one life less is displayed. Your implementation shows the total number of lives and greys them out instead of deleting them.
Ghost behavior lacks distinct modes (Scatter vs. Chase); the current implementation uses a weak movement bias that makes the game significantly easier than the original.
Ghost Pathfinding relies on a greedy distance minimization with significant randomness rather than a robust algorithm like BFS.
The "frightened mode" (blue ghosts) duration is far too short (60 steps) compared to the ALE, making the state barely observable
Small coin point values are incorrect (1 point instead of 10)
Power pellet position and amount differs
missing death animation
your game is missing the delay before movement is allowed on game start
the ghost "cage" in the center of the map is not implemented correctly, and killing ghosts does not trigger their respawn animation and their return to the cage. Instead, they immediately respawn at the upper right corner.

Implementation Quality: 3/4

Strengths

The step() function uses jax native functions, this is very good
The Game compiles very quickly
the code is understandable, especially because of the frequent and fitting comments

Weaknesses

The game only runs by using the “python scripts/play.py -g mspacman” command when deleting your “main” function and the imports above it. One Issue is that one of the imports imports the class that was written within the file itself ("from jaxatari.games.jax_mspacman import JaxMsPacman" in jax_mspacman.py). Also, including a main function with pygame logic inside a game implementation file violates the project's game structure. Refer to the JAXAtari Design Guide.
The reset function should be decorated with JIT
The Renderer uses if-else statements and for loops:
An Example is your render_text() function, which uses a 5x nested for-loop. This could be handled by a single sprite. Additionally, consider using vmaps more often.

Beamrider Team

tiago-hfs · 2026-01-07T14:47:09Z

Hi, your current implementation looks promising. The code looks mostly clean and it should be pretty straightforward to build on it to solve the tasks at hand.
That being said, there is still a lot of work to do to increase the similarity to the ALE, so here is my breakdown:

ALE-Similarity: 2/4
(I see, a lot of that was already mentioned in the previous comment. I think it was very thorough, but I found some more points that I will highlight)
Positives:

The general game principle is implemented:
- Ms. Pacman can move through the maze to collect pellets, which are added to the score.
- Collisions with ghosts cause deduction of a life and some points.
- The power pellet causes the ghosts to temporarily turn blue and be eatable.
The maze for level 1 looks correct in shape
The sprites for Ms. Pacman and the ghosts look correct, Ms. Pacman mouth animation looks correct

Negatives:

General:
- The game resets after 2000 steps. While it is fun to try and beat the game with a time limit, this should just be removed for further similarity.
- Levels after level 1 are missing
- The level has only 2 ghosts instead of 4
- The fruit never spawns
Speed: the movement speed of the player is too high compared to the ALE
Movement: The movement is not working as intended. Pressing a direction key outside of intersections should not lead Ms.Pacman to stop movement (and turn into the wrong direction)
Ghost movement:
- Ghosts make many random decisions, which leads to very hectic unfocused movement compared to the ALE.
- In the ALE there are two modes for movement: scatter and chase. Both the visual and action are not implemented
- Ghosts are only "eatable" for 60 steps, which is significantly shorter than the ALE
- Ghosts spawn in the corner, not the middle
Layout:
- color scheme and positioning of maze is different
- layout for points and deducted lives is different
- Your implementation also has a different logic for displaying lives than the ALE. You are displaying one more life than the ALE does, as it counts the current run as one life.
- The maze has way too many pellets, but too few power pellets
Point system:
- single pellet gives 1 point instead of 10
- eating the power pellet should grant 50 points
- eating a ghost gives 10 points instead of 200/400/800/1600 for the 1st/2nd/3rd/4th ghost during "frightened mode"

Implementation Quality: 3/4

Positives:

Mostly clean, readable code, well commented
Quick compilation
Good use of jax natives (cond, scan, select, vmap, dynamic_slice) for movement/rendering etc.
Modular design well adapted, respecting the JAXAtari Design Guide

Negatives:

Coding letters as lists of lists that have to be iterated through with nested loops for rendering is unnecessarily inefficient -> simple sprites can be used
There are still a bunch of for-loops that should preferably be replaced with jax alternatives or removed altogether. In particular there are multiple nested for loops in the init function of the JaxMsPacman class, which are seemingly redundant. (iterating over the maze layout twice)
Confusing use of 'G': used both as ghost spawn and power pellet position
The constant ghost_move_period is defined twice in MsPacmanConstants
Unlike the group before me, I was able to run the game normally without removing anything. That being said, the main() function should not be there as it violates the design guide (module should be pure and self-contained)

I wish you the best for the further development!

Best regards,
Tiago Soares

github-actions · 2026-01-08T18:01:29Z

📁 Previous CI results (run #20854526552)

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/20826471586

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-01-08 18:01:27 UTC.

github-actions · 2026-01-09T14:16:02Z

📁 Previous CI results (run #20855576681)

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/20854526552

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-01-09 14:16:00 UTC.

github-actions · 2026-01-09T14:53:59Z

📁 Previous CI results (run #22644030274)

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/20855576681

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-01-09 14:53:57 UTC.

… mechanics for ghosts (better ai) and rendering, added cherry behaviours (other fruits needed)

github-actions · 2026-03-03T21:46:04Z

📁 Previous CI results (run #22644202755)

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/22644030274

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:

mspacman ❌

============================= 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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[mspacman]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[mspacman]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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[mspacman]%0A%0ATypeError: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[mspacman]%0A%0ATypeError: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Per...*[Comment body truncated]*

</details>

PaulSeitz · 2026-03-06T12:42:08Z

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/22645158333

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:
mspacman ❌

I don't know why are the tests failing, error doesn't make much sense:
╭─    ~/Dev/Code/TUd/PAI/JAXAtari_D    MsPackmanBenni ?2      JAXAtari  3.11.14 via JAXAtari   11:39:17 PM ─╮
╰─❯ pytest --game mspacman --tb=line -ra tests/test_core_and_wrappers.py                                          ─╯
================================================ test session starts ================================================
platform darwin -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /Users/dimi/Developer/Code/TUd/PAI/JAXAtari_D
configfile: pyproject.toml
plugins: syrupy-4.9.1, jaxtyping-0.3.9
collected 10 items                                                                                                  

tests/test_core_and_wrappers.py ..........                                                                    [100%]

================================================ 10 passed in 13.49s ================================================
Hey, have you synced your fork with the current dev / master changes yet? Because GitHub 'merges' your branch and the dev branch for the tests and then you can pass the tests locally but have failed tests in GitHub actions.
Hi, I haven't synced dev with this brach, though i did sync the fork some time ago. I just had a look at the pipeline and it seems it cuts the src/test code from the origin and run that, thanks for the tip :)

No problem, I think specifically it fails because your renderer does not have the 'config' argument that is newly introduced and because the datatypes of state, obs and info are now standardized and I think yours are currently different. Should all be simple refactors, sync your fork and check some of the other envs for reference.

On another note, have you sent your mod ideas to someone yet? Because according to our list we have not received them?

DimiM99 · 2026-03-06T12:43:52Z

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/22645158333

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:
mspacman ❌

I don't know why are the tests failing, error doesn't make much sense:
╭─    ~/Dev/Code/TUd/PAI/JAXAtari_D    MsPackmanBenni ?2      JAXAtari  3.11.14 via JAXAtari   11:39:17 PM ─╮
╰─❯ pytest --game mspacman --tb=line -ra tests/test_core_and_wrappers.py                                          ─╯
================================================ test session starts ================================================
platform darwin -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /Users/dimi/Developer/Code/TUd/PAI/JAXAtari_D
configfile: pyproject.toml
plugins: syrupy-4.9.1, jaxtyping-0.3.9
collected 10 items                                                                                                  

tests/test_core_and_wrappers.py ..........                                                                    [100%]

================================================ 10 passed in 13.49s ================================================
Hey, have you synced your fork with the current dev / master changes yet? Because GitHub 'merges' your branch and the dev branch for the tests and then you can pass the tests locally but have failed tests in GitHub actions.
Hi, I haven't synced dev with this brach, though i did sync the fork some time ago. I just had a look at the pipeline and it seems it cuts the src/test code from the origin and run that, thanks for the tip :)
No problem, I think specifically it fails because your renderer does not have the 'config' argument that is newly introduced and because the datatypes of state, obs and info are now standardized and I think yours are currently different. Should all be simple refactors, sync your fork and check some of the other envs for reference.

On another note, have you sent your mod ideas to someone yet? Because according to our list we have not received them?

Yeah, i've seen the output in the pipeline but the version I had didn't have that 😅 though i was crazy. Thanks again for taking a look

github-actions · 2026-03-06T12:45:21Z

📁 Previous CI results (run #22808478665)

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/22763878272

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:

mspacman ❌

============================= 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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[mspacman]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[mspacman]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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[mspacman]%0A%0ATypeError: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[mspacman]%0A%0ATypeError: MsPacmanRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=244::Per...*[Comment body truncated]*

</details>

DimiM99 · 2026-03-07T20:11:50Z

I was on the way to update the code base for the version of the source changes, those were breaking.

I did pip install -e ".[dev]" but then the I got this

RuntimeError: 
❌  OWNERSHIP NOT CONFIRMED
----------------------------------------------------
To use JaxAtari, you must confirm ownership of the original 
 Atari 2600 ROMs and download the required assets.

Please run the following command in your terminal:

    .venv/bin/install-sprites

but when doing that:

╭─    ~/Developer/Code/TUd/PAI/JAXAtari_D    MsPackmanBenni *1 +1 !1 ?2                           3.10.19 via JAXAtari   09:03:03 PM ─╮
╰─❯ python -m jaxatari.install_sprites --confirm-ownership                                                                                 ─╯


OWNERSHIP CONFIRMATION
------------------------------------------
I declare to legally own a license to the original Atari 2600 ROMs.
I agree to not distribute these extracted game sprites and wish to proceed.

Do you confirm ownership ? [y/N]:

while I'm fine on the second thing, I definitely do not own the license thingies. How thin of ice are we walking here if i say yes to these? @PaulSeitz

github-actions · 2026-03-07T22:26:52Z

📁 Previous CI results (run #22810796198)

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/22808478665

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:

mspacman ❌

============================= 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]

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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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[mspacman]%0A%0AAssertionError: Renderer config was not updated! It is None. Wrapper might be falling back to slow resizing.%0Aassert None == (84, 84)%0A +  where None = RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None).downscale%0A +    where RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None) = <jax_mspacman.MsPacmanRenderer object at 0x7fc9300facd0>.config%0A +      where <jax_mspacman.MsPacmanRenderer object at 0x7fc9300facd0> = <jax_mspacman.JaxMsPacman object at 0x7fc926f99950>.renderer
::error file=tests/test_core_and_wrappers.py,line=493::test_native_downscaling_hot_swap[mspacman]%0A%0AAssertionError: Renderer config was not updated! It is None. Wrapper might be falling back to slow resizing.%0Aassert None == (84, 84)%0A +  where None = RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None).downscale%0A +    where RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None) = <jax_mspacman.MsPacmanRenderer object at 0x7fc9300facd0>.config%0A +      where <jax_mspacman.MsPacmanRenderer object at 0x7fc9300facd0> = <jax_mspacman.JaxMsPacman object at 0x7fc926f99950>.renderer
F::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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[mspacman]%0A%0AAssertionError: Renderer config channels should be 1 for grayscale%0Aassert 3 == 1%0A +  where 3 = RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None).channels%0A +    where RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None) = <jax_mspacman.MsPacmanRenderer object at 0x7fc926733010>.config%0A +      where <jax_mspacman.MsPacmanRenderer object at 0x7fc926733010> = <jax_mspacman.JaxMsPacman object at 0x7fc9267a06d0>.renderer
::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[mspacman]%0A%0AAssertionError: Renderer config channels should be 1 for grayscale%0Aassert 3 == 1%0A +  where 3 = RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None).channels%0A +    where RendererConfig(game_dimensions=(210, 160), channels=3, downscale=None) = <jax_mspacman.MsPacmanRenderer object at 0x7fc926733010>.config%0A +      where <jax_mspacman.MsPacmanRenderer object at 0x7fc926733010> = <jax_mspacman.JaxMsPacman object at 0x7fc9267a06d0>.renderer
F::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing 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_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_mspacman.py,line=400::Performance Warning: JaxMsPacman.consts is a 'NamedTuple'. This pre...*[Comment body truncated]*

</details>

github-actions · 2026-03-08T00:59:23Z

📁 Previous CI results (run #22824057952)

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/22810796198

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-08 00:59:21 UTC.

PaulSeitz · 2026-03-08T14:10:42Z

I was on the way to update the code base for the version of the source changes, those were breaking.

I did pip install -e ".[dev]" but then the I got this

RuntimeError: 
❌  OWNERSHIP NOT CONFIRMED
----------------------------------------------------
To use JaxAtari, you must confirm ownership of the original 
 Atari 2600 ROMs and download the required assets.

Please run the following command in your terminal:

    .venv/bin/install-sprites

but when doing that:

╭─    ~/Developer/Code/TUd/PAI/JAXAtari_D    MsPackmanBenni *1 +1 !1 ?2                           3.10.19 via JAXAtari   09:03:03 PM ─╮
╰─❯ python -m jaxatari.install_sprites --confirm-ownership                                                                                 ─╯


OWNERSHIP CONFIRMATION
------------------------------------------
I declare to legally own a license to the original Atari 2600 ROMs.
I agree to not distribute these extracted game sprites and wish to proceed.

Do you confirm ownership ? [y/N]:

while I'm fine on the second thing, I definitely do not own the license thingies. How thin of ice are we walking here if i say yes to these? @PaulSeitz

The check is just a legal safeguard to ensure we don't run into any issues with distributing 'original' game sprites. Since you are only developing locally and not distributing the files publicly, you should be good in accepting the prompt. Please also check our moodle announcement: https://moodle.informatik.tu-darmstadt.de/mod/forum/discuss.php?d=78886

github-actions · 2026-03-08T15:33:24Z

📁 Previous CI results (run #22895587838)

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/22824057952

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-08 15:33:22 UTC.

github-actions · 2026-03-10T09:26:18Z

📁 Previous CI results (run #22900318045)

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/22895587838

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:

mspacman ❌

============================= 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]

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
ssssssssssss.....s.::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[mspacman]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)%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=555::test_native_downscaling_grayscale[mspacman]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)%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_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[mspacman]%0A%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[mspacman]%0A%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
F.::error file=tests/test_environment_compatibility.py,line=115::TestBasicAPI.test_reset[mspacman]%0A%0AAssertionError: Reset observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array(19, dtype=int32), y=Array(25, dtype=int32), width=Array(1, dtype=int..., 0, 0],\n       [0, 0, 0, ..., 0, 0, 0]], dtype=int32), power_timer=Array(0, dtype=int32), lives=Array(3, dtype=int32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff526f65f10>, y: <jaxatari.spaces.Box object at 0x7ff526fdc4d0>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff526f65450>, lives: <jaxatari.spaces.Box object at 0x7ff526f66750>).contains
::error file=tests/test_environment_compatibility.py,line=115::TestBasicAPI.test_reset[mspacman]%0A%0AAssertionError: Reset observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array(19, dtype=int32), y=Array(25, dtype=int32), width=Array(1, dtype=int..., 0, 0],\n       [0, 0, 0, ..., 0, 0, 0]], dtype=int32), power_timer=Array(0, dtype=int32), lives=Array(3, dtype=int32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff526f65f10>, y: <jaxatari.spaces.Box object at 0x7ff526fdc4d0>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff526f65450>, lives: <jaxatari.spaces.Box object at 0x7ff526f66750>).contains
F::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[mspacman]%0A%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[mspacman]%0A%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
F::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[mspacman]%0A%0AAssertionError: Step observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array(19, dtype=int32), y=Array(25, dtype=int32), width=Array(1, dtype=int..., 0, 0],\n       [0, 0, 0, ..., 0, 0, 0]], dtype=int32), power_timer=Array(0, dtype=int32), lives=Array(3, dtype=int32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff527b30b50>, y: <jaxatari.spaces.Box object at 0x7ff5267ec810>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff5243184d0>, lives: <jaxatari.spaces.Box object at 0x7ff524318e90>).contains
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[mspacman]%0A%0AAssertionError: Step observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array(19, dtype=int32), y=Array(25, dtype=int32), width=Array(1, dtype=int..., 0, 0],\n       [0, 0, 0, ..., 0, 0, 0]], dtype=int32), power_timer=Array(0, dtype=int32), lives=Array(3, dtype=int32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff527b30b50>, y: <jaxatari.spaces.Box object at 0x7ff5267ec810>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff5243184d0>, lives: <jaxatari.spaces.Box object at 0x7ff524318e90>).contains
F....::error file=tests/test_core_and_wrappers.py,line=183::test_multi_reward_log_wrapper[mspacman]%0A%0AAssertionError: Expected object shape (4, 1782), got (4, 1702)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
::error file=tests/test_core_and_wrappers.py,line=183::test_multi_reward_log_wrapper[mspacman]%0A%0AAssertionError: Expected object shape (4, 1782), got (4, 1702)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array([19, 19, 19, 19], dtype=int32), y=Array([25, 25, 25, 25], dtype=int3...0, ..., 0, 0, 0]]], dtype=int32), power_timer=Array([0, 0, 0, 0], dtype=int32), lives=Array([3, 3, 3, 3], dtype=int32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff5241517d0>, y: <jaxatari.spaces.Box object at 0x7ff526b4c790>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff526b3e8d0>, lives: <jaxatari.spaces.Box object at 0x7ff51dcd4410>).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array([19, 19, 19, 19], dtype=int32), y=Array([25, 25, 25, 25], dtype=int3...0, ..., 0, 0, 0]]], dtype=int32), power_timer=Array([0, 0, 0, 0], dtype=int32), lives=Array([3, 3, 3, 3], dtype=int32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff5241517d0>, y: <jaxatari.spaces.Box object at 0x7ff526b4c790>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff526b3e8d0>, lives: <jaxatari.spaces.Box object at 0x7ff51dcd4410>).contains
F::error file=tests/test_core_and_wrappers.py,line=254::test_flatten_observation_wrapper[mspacman]%0A%0Aassert 6808 == (4 * np.float64(1782.0))%0A +  where np.float64(1782.0) = get_object_centric_obs_size(Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f51b00a6390>, y: <jaxatari.spaces.Box object at 0x7f51a8fae610>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f51a8f34dd0>, lives: <jaxatari.spaces.Box object at 0x7f51a8f37490>))%0A +    where Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f51b00a6390>, y: <jaxatari.spaces.Box object at 0x7f51a8fae610>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f51a8f34dd0>, lives: <jaxatari.spaces.Box object at 0x7f51a8f37490>) = observation_space()%0A +      where observation_space = <jax_mspacman.JaxMsPacman object at 0x7f51b0ba07d0>.observation_space
::error file=tests/test_core_and_wrappers.py,line=254::test_flatten_observation_wrapper[mspacman]%0A%0Aassert 6808 == (4 * np.float64(1782.0))%0A +  where np.float64(1782.0) = get_object_centric_obs_size(Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f51b00a6390>, y: <jaxatari.spaces.Box object at 0x7f51a8fae610>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f51a8f34dd0>, lives: <jaxatari.spaces.Box object at 0x7f51a8f37490>))%0A +    where Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f51b00a6390>, y: <jaxatari.spaces.Box object at 0x7f51a8fae610>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f51a8f34dd0>, lives: <jaxatari.spaces.Box object at 0x7f51a8f37490>) = observation_space()%0A +      where observation_space = <jax_mspacman.JaxMsPacman object at 0x7f51b0ba07d0>.observation_space
F..::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([[19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32))%0A +    where contains = <jaxatari.spaces.Box object at 0x7ff51cf4ca50>.contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([[19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32))%0A +    where contains = <jaxatari.spaces.Box object at 0x7ff51cf4ca50>.contains
F::error file=tests/test_core_and_wrappers.py,line=345::test_flatten_observation_wrapper_space_structure[mspacman]%0A%0AAssertionError: Observation leaf with shape (6720,) not contained in space with shape (7040,)%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([0., 0., 0., ..., 0., 0., 0.], dtype=float32))%0A +    where contains = <jaxatari.spaces.Box object at 0x7f51b064f490>.contains
::error file=tests/test_core_and_wrappers.py,line=345::test_flatten_observation_wrapper_space_structure[mspacman]%0A%0AAssertionError: Observation leaf with shape (6720,) not contained in space with shape (7040,)%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([0., 0., 0., ..., 0., 0., 0.], dtype=float32))%0A +    where contains = <jaxatari.spaces.Box object at 0x7f51b064f490>.contains
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[  0,   0,   0],\n         [  0,   0,   0],\n         [  0,   0,   0],\n         ...,\n         [  0,   0,   0],...0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32)))%0A +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7ff51ddb6f90>, <jaxatari.spaces.Box object at 0x7ff51cfb9a50>).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[  0,   0,   0],\n         [  0,   0,   0],\n         [  0,   0,   0],\n         ...,\n         [  0,   0,   0],...0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32)))%0A +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7ff51ddb6f90>, <jaxatari.spaces.Box object at 0x7ff51cfb9a50>).contains
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([19., 25.,  1., ...,  0.,  0.,  3.], dtype=float32))%0A +    where contains = <jaxatari.spaces.Box object at 0x7ff51d2d5790>.contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([19., 25.,  1., ...,  0.,  0.,  3.], dtype=float32))%0A +    where contains = <jaxatari.spaces.Box object at 0x7ff51d2d5790>.contains
F..::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[  0,   0,   0],\n         [  0,   0,   0],\n         [  0,   0,   0],\n         ...,\n         [  0,   0,   0],...0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32)))%0A +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7f5195e40210>, <jaxatari.spaces.Box object at 0x7f5195f79810>).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[  0,   0,   0],\n         [  0,   0,   0],\n         [  0,   0,   0],\n         ...,\n         [  0,   0,   0],...0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32)))%0A +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7f5195e40210>, <jaxatari.spaces.Box object at 0x7f5195f79810>).contains
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([0., 0., 0., ..., 0., 0., 0.], dtype=float32), Array([19., 25.,  1., ...,  0.,  0.,  3.], dtype=float32)))%0A +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7ff51c7c8610>, <jaxatari.spaces.Box object at 0x7ff51c0612d0>).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-mspacman]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([0., 0., 0., ..., 0., 0., 0.], dtype=float32), Array([19., 25.,  1., ...,  0.,  0.,  3.], dtype=float32)))%0A +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7ff51c7c8610>, <jaxatari.spaces.Box object at 0x7ff51c0612d0>).contains
F.....::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-mspacman]%0A%0AAssertionError: Observation shape (4, 1702) should match space shape (4, 1782)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-mspacman]%0A%0AAssertionError: Observation shape (4, 1702) should match space shape (4, 1782)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
F.::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-mspacman]%0A%0AAssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-mspacman]%0A%0AAssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-mspacman]%0A%0AAssertionError: Observation shape (6808,) should match space shape (7128,)%0Aassert (6808,) == (7128,)%0A  %0A  At index 0 diff: 6808 != 7128%0A  %0A  Full diff:%0A    (%0A  -     7128,%0A  +     6808,%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-mspacman]%0A%0AAssertionError: Observation shape (6808,) should match space shape (7128,)%0Aassert (6808,) == (7128,)%0A  %0A  At index 0 diff: 6808 != 7128%0A  %0A  Full diff:%0A    (%0A  -     7128,%0A  +     6808,%0A    )
F..::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-mspacman]%0A%0AAssertionError: Observation part shape (6808,) should match space shape (7128,)%0Aassert (6808,) == (7128,)%0A  %0A  At index 0 diff: 6808 != 7128%0A  %0A  Full diff:%0A    (%0A  -     7128,%0A  +     6808,%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-mspacman]%0A%0AAssertionError: Observation part shape (6808,) should match space shape (7128,)%0Aassert (6808,) == (7128,)%0A  %0A  At index 0 diff: 6808 != 7128%0A  %0A  Full diff:%0A    (%0A  -     7128,%0A  +     6808,%0A    )
F.::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-mspacman]%0A%0AAssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-mspacman]%0A%0AAssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)%0Aassert (4, 1702) == (4, 1782)%0A  %0A  At index 1 diff: 1702 != 1782%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     1782,%0A  ?       ^%0A  +     1702,%0A  ?       ^%0A    )
F........ [ 50%]
...........ssssssssssssssss::error file=tests/test_environment_compatibility.py,line=593::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[mspacman]%0A%0AAssertionError: Flattened observation should be contained in space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array([19., 19., 19., 19.], dtype=float32), y=Array([25., 25., 25., 25.], ... 0.], dtype=float32), power_timer=Array([0., 0., 0., 0.], dtype=float32), lives=Array([3., 3., 3., 3.], dtype=float32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f517d7fa2d0>, y: <jaxatari.spaces.Box object at 0x7f5208081590>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f517d73de10>, lives: <jaxatari.spaces.Box object at 0x7f517d610b90>).contains
::error file=tests/test_environment_compatibility.py,line=593::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[mspacman]%0A%0AAssertionError: Flattened observation should be contained in space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array([19., 19., 19., 19.], dtype=float32), y=Array([25., 25., 25., 25.], ... 0.], dtype=float32), power_timer=Array([0., 0., 0., 0.], dtype=float32), lives=Array([3., 3., 3., 3.], dtype=float32)))%0A +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f517d7fa2d0>, y: <jaxatari.spaces.Box object at 0x7f5208081590>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f517d73de10>, lives: <jaxatari.spaces.Box object at 0x7f517d610b90>).contains
F..::error file=tests/test_environment_compatibility.py,line=640::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[mspacman]%0A%0ATypeError: sub got incompatible shapes for broadcasting: (4, 1702), (4, 1782).%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=640::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[mspacman]%0A%0ATypeError: sub got incompatible shapes for broadcasting: (4, 1702), (4, 1782).%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/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::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/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::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/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
.::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
.::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
..::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
..::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
s.........  [100%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/numpy/util.py:277: ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:97: assert (4, 1702) == (4, 1782)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:126: AssertionError: Reset observation should be contained in observation space
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:124: assert (4, 1702) == (4, 1782)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:165: AssertionError: Step observation should be contained in observation space
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:216: AssertionError: Expected object shape (4, 1782), got (4, 1702)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:268: assert 6808 == (4 * np.float64(1782.0))
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:429: AssertionError: Observation leaf with shape (6720,) not contained in space with shape (7040,)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:372: AssertionError: Observation shape (4, 1702) should match space shape (4, 1782)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:353: AssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:372: AssertionError: Observation shape (6808,) should match space shape (7128,)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:353: AssertionError: Observation part shape (6808,) should match space shape (7128,)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:353: AssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:622: AssertionError: Flattened observation should be contained in space
/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/numpy/ufuncs.py:1569: TypeError: sub got incompatible shapes for broadcasting: (4, 1702), (4, 1782).
=============================== 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_render_method[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_reset_method[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_preprocessing_wrappers[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_frame_stack_wrapper[mspacman]
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/gym_wrapper.py:131: UserWarning: Environment returned a NamedTuple for 'info'. This is deprecated.
    warnings.warn(

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[mspacman]
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/gym_wrapper.py:111: UserWarning: Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
    warnings.warn(

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[mspacman]
  /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[mspacman]
  /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(

-- 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:671: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:694: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:542: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'mspacman' is not in core.GAME_MODULES
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
FAILED tests/test_core_and_wrappers.py::test_native_downscaling_grayscale[mspacman] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_pixel_and_object_centric_wrapper[mspacman] - assert (4, 1702) == (4, 1782)
  
  At index 1 diff: 1702 != 1782
  
  Full diff:
    (
        4,
  -     1782,
  ?       ^
  +     1702,
  ?       ^
    )
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_reset[mspacman] - AssertionError: Reset observation should be contained in observation space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array(19, dtype=int32), y=Array(25, dtype=int32), width=Array(1, dtype=int..., 0, 0],\n       [0, 0, 0, ..., 0, 0, 0]], dtype=int32), power_timer=Array(0, dtype=int32), lives=Array(3, dtype=int32)))
 +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff526f65f10>, y: <jaxatari.spaces.Box object at 0x7ff526fdc4d0>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff526f65450>, lives: <jaxatari.spaces.Box object at 0x7ff526f66750>).contains
FAILED tests/test_core_and_wrappers.py::test_object_centric_wrapper[mspacman] - assert (4, 1702) == (4, 1782)
  
  At index 1 diff: 1702 != 1782
  
  Full diff:
    (
        4,
  -     1782,
  ?       ^
  +     1702,
  ?       ^
    )
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_step[mspacman] - AssertionError: Step observation should be contained in observation space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array(19, dtype=int32), y=Array(25, dtype=int32), width=Array(1, dtype=int..., 0, 0],\n       [0, 0, 0, ..., 0, 0, 0]], dtype=int32), power_timer=Array(0, dtype=int32), lives=Array(3, dtype=int32)))
 +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff527b30b50>, y: <jaxatari.spaces.Box object at 0x7ff5267ec810>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff5243184d0>, lives: <jaxatari.spaces.Box object at 0x7ff524318e90>).contains
FAILED tests/test_core_and_wrappers.py::test_multi_reward_log_wrapper[mspacman] - AssertionError: Expected object shape (4, 1782), got (4, 1702)
assert (4, 1702) == (4, 1782)
  
  At index 1 diff: 1702 != 1782
  
  Full diff:
    (
        4,
  -     1782,
  ?       ^
  +     1702,
  ?       ^
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Atari-mspacman] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array([19, 19, 19, 19], dtype=int32), y=Array([25, 25, 25, 25], dtype=int3...0, ..., 0, 0, 0]]], dtype=int32), power_timer=Array([0, 0, 0, 0], dtype=int32), lives=Array([3, 3, 3, 3], dtype=int32)))
 +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7ff5241517d0>, y: <jaxatari.spaces.Box object at 0x7ff526b4c790>...0>, power_timer: <jaxatari.spaces.Box object at 0x7ff526b3e8d0>, lives: <jaxatari.spaces.Box object at 0x7ff51dcd4410>).contains
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper[mspacman] - assert 6808 == (4 * np.float64(1782.0))
 +  where np.float64(1782.0) = get_object_centric_obs_size(Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f51b00a6390>, y: <jaxatari.spaces.Box object at 0x7f51a8fae610>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f51a8f34dd0>, lives: <jaxatari.spaces.Box object at 0x7f51a8f37490>))
 +    where Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f51b00a6390>, y: <jaxatari.spaces.Box object at 0x7f51a8fae610>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f51a8f34dd0>, lives: <jaxatari.spaces.Box object at 0x7f51a8f37490>) = observation_space()
 +      where observation_space = <jax_mspacman.JaxMsPacman object at 0x7f51b0ba07d0>.observation_space
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[ObjectCentric-mspacman] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(Array([[19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32))
 +    where contains = <jaxatari.spaces.Box object at 0x7ff51cf4ca50>.contains
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper_space_structure[mspacman] - AssertionError: Observation leaf with shape (6720,) not contained in space with shape (7040,)
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(Array([0., 0., 0., ..., 0., 0., 0.], dtype=float32))
 +    where contains = <jaxatari.spaces.Box object at 0x7f51b064f490>.contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[PixelAndObjectCentric-mspacman] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains((Array([[[[  0,   0,   0],\n         [  0,   0,   0],\n         [  0,   0,   0],\n         ...,\n         [  0,   0,   0],...0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32)))
 +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7ff51ddb6f90>, <jaxatari.spaces.Box object at 0x7ff51cfb9a50>).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[FlattenedObjectCentric-mspacman] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(Array([19., 25.,  1., ...,  0.,  0.,  3.], dtype=float32))
 +    where contains = <jaxatari.spaces.Box object at 0x7ff51d2d5790>.contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[MultiRewardLogged-mspacman] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains((Array([[[[  0,   0,   0],\n         [  0,   0,   0],\n         [  0,   0,   0],\n         ...,\n         [  0,   0,   0],...0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.],\n       [19., 25.,  1., ...,  0.,  0.,  3.]], dtype=float32)))
 +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7f5195e40210>, <jaxatari.spaces.Box object at 0x7f5195f79810>).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-mspacman] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains((Array([0., 0., 0., ..., 0., 0., 0.], dtype=float32), Array([19., 25.,  1., ...,  0.,  0.,  3.], dtype=float32)))
 +    where contains = Tuple(<jaxatari.spaces.Box object at 0x7ff51c7c8610>, <jaxatari.spaces.Box object at 0x7ff51c0612d0>).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[ObjectCentric-mspacman] - AssertionError: Observation shape (4, 1702) should match space shape (4, 1782)
assert (4, 1702) == (4, 1782)
  
  At index 1 diff: 1702 != 1782
  
  Full diff:
    (
        4,
  -     1782,
  ?       ^
  +     1702,
  ?       ^
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[PixelAndObjectCentric-mspacman] - AssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)
assert (4, 1702) == (4, 1782)
  
  At index 1 diff: 1702 != 1782
  
  Full diff:
    (
        4,
  -     1782,
  ?       ^
  +     1702,
  ?       ^
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[FlattenedObjectCentric-mspacman] - AssertionError: Observation shape (6808,) should match space shape (7128,)
assert (6808,) == (7128,)
  
  At index 0 diff: 6808 != 7128
  
  Full diff:
    (
  -     7128,
  +     6808,
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[LoggedFlattenedPixelAndObject-mspacman] - AssertionError: Observation part shape (6808,) should match space shape (7128,)
assert (6808,) == (7128,)
  
  At index 0 diff: 6808 != 7128
  
  Full diff:
    (
  -     7128,
  +     6808,
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[MultiRewardLogged-mspacman] - AssertionError: Observation part shape (4, 1702) should match space shape (4, 1782)
assert (4, 1702) == (4, 1782)
  
  At index 1 diff: 1702 != 1782
  
  Full diff:
    (
        4,
  -     1782,
  ?       ^
  +     1702,
  ?       ^
    )
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_flatten_observation_wrapper_space_structure[mspacman] - AssertionError: Flattened observation should be contained in space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(MsPacmanObservation(pacman=EntityPosition(x=Array([19., 19., 19., 19.], dtype=float32), y=Array([25., 25., 25., 25.], ... 0.], dtype=float32), power_timer=Array([0., 0., 0., 0.], dtype=float32), lives=Array([3., 3., 3., 3.], dtype=float32)))
 +    where contains = Dict(pacman: Dict(x: <jaxatari.spaces.Box object at 0x7f517d7fa2d0>, y: <jaxatari.spaces.Box object at 0x7f5208081590>...0>, power_timer: <jaxatari.spaces.Box object at 0x7f517d73de10>, lives: <jaxatari.spaces.Box object at 0x7f517d610b90>).contains
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_normalize_observation_wrapper[mspacman] - TypeError: sub got incompatible shapes for broadcasting: (4, 1702), (4, 1782).
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
====== 21 failed, 86 passed, 36 skipped, 15 warnings in 171.26s (0:02:51) ======

This log was automatically created at 2026-03-10 09:26:16 UTC.

github-actions · 2026-03-10T11:38:30Z

📁 Previous CI results (run #22901500963)

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/22900318045

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:

mspacman ❌

============================= 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]

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
ssssssssssss.....s.::error file=tests/test_core_and_wrappers.py,line=555::test_native_downscaling_grayscale[mspacman]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)%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=555::test_native_downscaling_grayscale[mspacman]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)%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............................................. [ 50%]
..............ssssssssssssssss.......................::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::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=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::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/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::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/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
.::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
..::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
.::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
.::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=111::Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
.::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
::warning file=src/jaxatari/gym_wrapper.py,line=131::Environment returned a NamedTuple for 'info'. This is deprecated.
s..........  [100%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/numpy/util.py:277: ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)
=============================== 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[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_render_method[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_reset_method[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_preprocessing_wrappers[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_frame_stack_wrapper[mspacman]
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/gym_wrapper.py:131: UserWarning: Environment returned a NamedTuple for 'info'. This is deprecated.
    warnings.warn(

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[mspacman]
tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[mspacman]
tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[mspacman]
  /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/gym_wrapper.py:111: UserWarning: Environment returned a NamedTuple for 'info'. This is deprecated. Please return a Dict or a Flax PyTreeNode.
    warnings.warn(

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[mspacman]
  /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[mspacman]
  /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(

-- 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:671: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:694: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:542: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'mspacman' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'mspacman' is not in core.GAME_MODULES
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
FAILED tests/test_core_and_wrappers.py::test_native_downscaling_grayscale[mspacman] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (84, 1)
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
====== 1 failed, 106 passed, 36 skipped, 15 warnings in 213.64s (0:03:33) ======

This log was automatically created at 2026-03-10 11:38:28 UTC.

github-actions · 2026-03-10T12:06:48Z

📁 Previous CI results (run #22945937510)

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/22901500963

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-10 12:06:46 UTC.

github-actions · 2026-03-11T09:37:49Z

📁 Previous CI results (run #22950798186)

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/22945937510

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-11 09:37:47 UTC.

github-actions · 2026-03-11T11:45:23Z

📁 Previous CI results (run #22999793914)

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/22950798186

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-11 11:45:21 UTC.

github-actions · 2026-03-12T11:34:56Z

📁 Previous CI results (run #23447680343)

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/22999793914

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-12 11:34:54 UTC.

PaulSeitz · 2026-03-19T00:54:08Z

Hello, just a quick question about your modifications. Have you written (either here or per mail) to any of the supervisors about which mods you are implementing for your environment (as asked for recently here: https://moodle.informatik.tu-darmstadt.de/mod/forum/discuss.php?d=78826)?

In our internal document we dont have your mods noted down and I want to make sure that we didnt overlook something.

github-actions · 2026-03-23T16:21:50Z

📁 Previous CI results (run #23448330361)

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/23447680343

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:

scripts/benchmarks/config/alg/ppo_jaxatari_mspacman_object.yaml
scripts/benchmarks/config/alg/pqn_jaxatari_mspacman_object.yaml
scripts/benchmarks/config/alg/pqn_jaxatari_mspacman_pixel.yaml

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-03-23 16:21:49 UTC.

github-actions · 2026-03-23T16:35:50Z

📁 Previous CI results (run #23449096818)

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/23448330361

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:

scripts/benchmarks/config/alg/ppo_jaxatari_mspacman_object.yaml
scripts/benchmarks/config/alg/pqn_jaxatari_mspacman_object.yaml
scripts/benchmarks/config/alg/pqn_jaxatari_mspacman_pixel.yaml
scripts/benchmarks/train_mspacman.sh

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-03-23 16:35:48 UTC.

github-actions · 2026-03-23T16:52:39Z

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/23449096818

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:

scripts/benchmarks/config/alg/ppo_jaxatari_mspacman_object.yaml
scripts/benchmarks/config/alg/pqn_jaxatari_mspacman_object.yaml
scripts/benchmarks/config/alg/pqn_jaxatari_mspacman_pixel.yaml
scripts/benchmarks/train_mspacman.sh

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-03-23 16:52:37 UTC.

benninho21 and others added 10 commits November 11, 2025 14:46

Basic structure of the code and mechanics

f79b140

Basis is set with mechanics and basic maze and ghoests

57a3c31

Functionality works all right and can be launched from the play.py sc…

7a1db0d

…ript TODOgit add .! Tests and visuals

adding sprites, and extentding logic for it

e8f21eb

reworking the maze rendering, adapting and extending game logic

ffc7395

Logic works the pallets are kind of broken

8b28515

visual edits/extentions:

4a9b5c2

- sprites for mx pac man ghosts - pallets

added pre and post game screens, implemented 3 lives logic

b613935

ui and hearts style fixes

7764df3

resolve conflicts

a352d6b

removing unwanted change (hope I don't have to rebase)

e143684

DimiM99 changed the title ~~Add Ms. Pac-Man Game to JAXAtari~~ First Submission: Ms. Pac-Man Dec 21, 2025

PaulSeitz self-assigned this Jan 6, 2026

Scoreboard fixed

88052f0

number of pellets is corrected

2f26a16

Pellets sice is done visually

e81b6a5

DimiM99 added 2 commits March 3, 2026 22:39

update/add new sprites with animations frames

ee14a59

update for feedback, completing (close to) perfect level 1. Addapting…

fb0d41b

… mechanics for ghosts (better ai) and rendering, added cherry behaviours (other fruits needed)

removing unnecessary frames

6a9b39b

added levels and advancement of levels

6d61369

code refactor, frameqwork tests should pass now

a32a805

pellets and sprites added

aa577e7

Level and visual fix

6bea9f9

Visuals done

9807e90

pipeline fix

3db7646

bugfix level deaths

7abfa6b

Fruitpath added

90dc130

Ghost bug fixed

d9e3de7

adding the config files

7e4752e

adding training runner script

fdae5f9

adj script

26b37d2

Conversation

DimiM99 commented Dec 21, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

🎮 Add Ms. Pac-Man Game to JAXAtari

Summary

✅ Features Implemented

📁 New Files Added

🎮 How to Play

🏗️ Architecture

👨‍💻👨‍💻 Team

Uh oh!

github-actions Bot commented Dec 21, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ✅

Uh oh!

github-actions Bot commented Dec 21, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ✅

Uh oh!

github-actions Bot commented Dec 21, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ✅

Uh oh!

github-actions Bot commented Dec 21, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ✅

Framework Tests ✅

Uh oh!

github-actions Bot commented Dec 21, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ✅

Framework Tests ✅

Uh oh!

gixerl commented Dec 29, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

ALE-similarity: 1.5/4

Implementation Quality: 3/4

Uh oh!

tiago-hfs commented Jan 7, 2026

Uh oh!

github-actions Bot commented Jan 8, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ✅

Framework Tests ✅

Uh oh!

github-actions Bot commented Jan 9, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ✅

Framework Tests ✅

Uh oh!

github-actions Bot commented Jan 9, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ✅

Framework Tests ✅

Uh oh!

github-actions Bot commented Mar 3, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Test Report

Base Branch ✅

Changed Files ✅

Framework Tests ❌

DimiM99 commented Dec 21, 2025 •

edited

Loading

github-actions Bot commented Dec 21, 2025 •

edited

Loading

github-actions Bot commented Dec 21, 2025 •

edited

Loading

github-actions Bot commented Dec 21, 2025 •

edited

Loading

github-actions Bot commented Dec 21, 2025 •

edited

Loading

github-actions Bot commented Dec 21, 2025 •

edited

Loading

gixerl commented Dec 29, 2025 •

edited

Loading

github-actions Bot commented Jan 8, 2026 •

edited

Loading

github-actions Bot commented Jan 9, 2026 •

edited

Loading

github-actions Bot commented Jan 9, 2026 •

edited

Loading

github-actions Bot commented Mar 3, 2026 •

edited

Loading

DimiM99 commented Mar 6, 2026 •

edited

Loading

github-actions Bot commented Mar 6, 2026 •

edited

Loading

github-actions Bot commented Mar 7, 2026 •

edited

Loading

github-actions Bot commented Mar 8, 2026 •

edited

Loading

github-actions Bot commented Mar 8, 2026 •

edited

Loading

github-actions Bot commented Mar 10, 2026 •

edited

Loading

github-actions Bot commented Mar 10, 2026 •

edited

Loading

github-actions Bot commented Mar 10, 2026 •

edited

Loading