Feature/tictactoe3d

[imnikitk]

Hello there,
this is the implementation of TicTacToe3d. We have tried to replicate the exact TicTacToe3d by JAXAtari. We have currently only AI vs User implementation and we will be adding other mods in the near future. We needed to change some things in our project as given by the instructor since i have a intel mac which just wouldn't support jax and jaxlib 0.7.0. Here are the changes we had to make: pyproject.toml:

dependencies = [
"absl-py>=2.3",
"ale-py>=0.11.1",
"chex>=0.1.87",
"flax>=0.10.2,<0.11.0", <- changed
"gymnasium>=1.2.0",
"gymnax>=0.0.8",
"jax>=0.4.38,<0.5.0", <- changed (insert what works for you)
"ml-dtypes>=0.4.0,<0.5.0", <- changed
"numpy>=1.26.0,<2.0.0", <- changed
"opt-einsum>=3.4.0",
"scipy>=1.15.3",
"toolz>=1.0.0",
"typing-extensions>=4.14.0",
]

We look forward to the review and happy to change. Thank you.

[github-actions]

📁 Previous CI results (run #23595193613)

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

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:

• .gitignore
• TicTacToe3D_screenshots/background.npy
• TicTacToe3D_screenshots/cursor.npy
• TicTacToe3D_screenshots/frame_00001.npy
• TicTacToe3D_screenshots/frame_00002.npy
• TicTacToe3D_screenshots/frame_00003.npy
• TicTacToe3D_screenshots/o.npy
• TicTacToe3D_screenshots/x.npy
• docs/source/requirements.txt
• pyproject.toml
• testing_tictactoe3d.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 13:26:16 UTC.

[xray2moon]

Hey, we really liked your implementation. Great Job! :)

ALE-similarity: 3/4

Overall your implementation closely resembles the original game, but differs in a few instances.

Strengths:

• The Level layout is implemented correctly
• The game controls are implemented correctly. They are also more responsive than the actual ALE. However this causes a slight divergence from the actual ALE.

Differences:

• In your environment the player makes the first move, whereas in the ALE the enemy makes the first move.
• The enemy in the ALE is significantly stronger. Yours doesn’t seem to follow any offensive strategy, instead it seems to mainly prevent immediate attacks.
• Navigation in the original uses edge wraparound: leaving the grid in one direction re-enters from the opposite side. In this implementation, boundary crossings are blocked, so traversal across the grid deviates from the reference behavior.
• As soon as you win a round, you instantly place your player mark at the top left corner.
• The correct dimensions for the game are 210 x 160 instead of 207x156.

Visual differences:
These differences don’t change the gameplay but are still mentioned for completeness.

• The last enemy step is highlighted by making it blink in the ALE. In the environment this blinking effect is missing.
• In the ALE, after every step, the entire screen goes black. In your implementation that does not happen.
• While selecting a tile, the player blinks faster in the ALE than in your game. (The slower pace makes your implementation less stressful)

Implementation Quality: 4/4

Strengths

• This code is concise and achieves the entire logic with around 400 lines of code
• The logic is easy to understand
• Everything seems jittable

Weaknesses

• The line in 339 does not really make sense as it is logically redundant.
• Ideally the step function should implemented as an orchestrator that only calls helper functions. In your implementation the step function contains most of the step logic directly. Creating separate functions for player step and CPU step that compute parts of the state change such as it is implemented in Kangaroo would help modularity and modification development.

Beamrider Team

[DimiM99]

Hello there, I'm about to review your first submission, before that I have a tip for you on how to fix your jax and conda env issues:

• revert the changes made outside your game so the test can pass
• you can use conda/miniconda right there in your project use conda create -p .venv python=3.10 -y and it will get the env running for u right in the project directory and will be ignored by the git :) (I assume the conda folder in .gitignore is for that) or better install the env as a global one and activate by name (less pain in the behind if u ask me)
• intel macs do support the jax version of this project as long as you have an AMD gpu on board, in which case you'll need ti install jax-metal (latest is 0.1.0) plugin from pip and should work.

if you only have the intel GPU refer to this in which case you wont need the jax-metal plugin

[DimiM99]

Here are my thoughts on the first version of the game:

Visuals and ALE-Similarity (4/4)

there are some small differences but those actually raise the bar in my opinion on how the game behaves so i wouldn't penalize that.

a good change option:

• non-standard frame dimensions**

ALE TicTacToe3D-v5: (210, 160, 3)  ← Standard Atari resolution
JAXAtari:          (207, 156, 3)  ← Non-standard

This breaks compatibility with:

• Pre-trained models expecting 210x160 input
• Standard Atari preprocessing pipelines
• Direct frame comparison tools (which i had to debug a bit to get the image above)

Implementation Quality (3/4)

The game is fairly modular and readable (thanks for not including comments on every single thing) - i could easily find my way around it. Main issue (hense 1 missing point) is the better options for jax integration and bettering performance:

optimisation option:

• vectorize the render loop, instead of using python for loop over 64 cells with nested jax.lax.cond calls. It creates 128 conditional branches per frame, preventing efficient GPU parallelization.

I really liked:

• _compute_cpu_move() uses efficient vectorized jnp.tensordot for win detection
• _check_winner() is well-vectorized
• Win masks are pre-computed at module load time

[github-actions]

📁 Previous CI results (run #23618805813)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

pong ❌

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

......::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=694::TestDatatypeConsistency.test_constants_are_pytree_node[pong]%0A%0AAssertionError: Constants should be flax.struct.PyTreeNode, got <class 'jax_pong.PongConstants'>. Please refactor from NamedTuple to PyTreeNode.%0Aassert False
::error file=tests/test_all_mods.py,line=694::TestDatatypeConsistency.test_constants_are_pytree_node[pong]%0A%0AAssertionError: Constants should be flax.struct.PyTreeNode, got <class 'jax_pong.PongConstants'>. Please refactor from NamedTuple to PyTreeNode.%0Aassert False
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=712::TestDatatypeConsistency.test_state_is_struct_dataclass[pong]%0A%0AAssertionError: State should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongState'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=712::TestDatatypeConsistency.test_state_is_struct_dataclass[pong]%0A%0AAssertionError: State should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongState'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=732::TestDatatypeConsistency.test_observation_is_struct_dataclass[pong]%0A%0AAssertionError: Observation should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongObservation'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=732::TestDatatypeConsistency.test_observation_is_struct_dataclass[pong]%0A%0AAssertionError: Observation should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongObservation'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=752::TestDatatypeConsistency.test_info_is_struct_dataclass[pong]%0A%0AAssertionError: Info should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongInfo'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=752::TestDatatypeConsistency.test_info_is_struct_dataclass[pong]%0A%0AAssertionError: Info should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongInfo'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[pong]%0A%0AAssertionError: Observation type: <class 'jax_pong.PongObservation'>%0Aassert False
::error file=tests/test_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[pong]%0A%0AAssertionError: Observation type: <class 'jax_pong.PongObservation'>%0Aassert False
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
...::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-lazy_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-lazy_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-random_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-random_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-zero_score-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-zero_score-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=.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
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[pong]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[pong]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[pong]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[pong]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
. [ 47%]
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This p...*[Comment body truncated]*

</details>

[github-actions]

📁 Previous CI results (run #23687662924)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

pong ❌

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

.......::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=694::TestDatatypeConsistency.test_constants_are_pytree_node[pong]%0A%0AAssertionError: Constants should be flax.struct.PyTreeNode, got <class 'jax_pong.PongConstants'>. Please refactor from NamedTuple to PyTreeNode.%0Aassert False
::error file=tests/test_all_mods.py,line=694::TestDatatypeConsistency.test_constants_are_pytree_node[pong]%0A%0AAssertionError: Constants should be flax.struct.PyTreeNode, got <class 'jax_pong.PongConstants'>. Please refactor from NamedTuple to PyTreeNode.%0Aassert False
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=712::TestDatatypeConsistency.test_state_is_struct_dataclass[pong]%0A%0AAssertionError: State should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongState'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=712::TestDatatypeConsistency.test_state_is_struct_dataclass[pong]%0A%0AAssertionError: State should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongState'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_all_mods.py,line=732::TestDatatypeConsistency.test_observation_is_struct_dataclass[pong]%0A%0AAssertionError: Observation should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongObservation'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=732::TestDatatypeConsistency.test_observation_is_struct_dataclass[pong]%0A%0AAssertionError: Observation should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongObservation'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=752::TestDatatypeConsistency.test_info_is_struct_dataclass[pong]%0A%0AAssertionError: Info should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongInfo'>.%0Aassert (False or False)
::error file=tests/test_all_mods.py,line=752::TestDatatypeConsistency.test_info_is_struct_dataclass[pong]%0A%0AAssertionError: Info should be @struct.dataclass (PyTreeNode), got <class 'jax_pong.PongInfo'>.%0Aassert (False or False)
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::error file=tests/test_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[pong]%0A%0AAssertionError: Observation type: <class 'jax_pong.PongObservation'>%0Aassert False
::error file=tests/test_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[pong]%0A%0AAssertionError: Observation type: <class 'jax_pong.PongObservation'>%0Aassert False
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
...::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-lazy_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-lazy_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-random_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-random_enemy-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-zero_score-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[pong-zero_score-individual]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=.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
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/core.py,line=170::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.

::warning file=src/jaxatari/core.py,line=182::Environment exposes deprecated obs_to_flat_array(). Observations should now be flax.struct.dataclasses using ObjectObservation for objects or plain arrays for observations like lives, score, etc. Depending on legacy obs_to_flat_array might lead to unforseen issues with wrappers.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[pong]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[pong]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[pong]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[pong]%0A%0ATypeError: PongRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
. [ 47%]
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_pong.py,line=106::Performance Warning: JaxPong.consts ...*[Comment body truncated]*

</details>

[github-actions]

📁 Previous CI results (run #23710015499)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssss...ssssssss..............::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.

::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........... [ 91%]
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 (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[bankheist]
  /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[bankheist]
  /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 [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:521: 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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 362.25s (0:06:02) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 221.18s (0:03:41) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
...............s..... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ❌

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

sssssssssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
Fsss::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
Fs.....s..::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=189::test_multi_reward_log_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=260::test_flatten_observation_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=320::test_log_wrapper_with_flatten_observation[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=320::test_log_wrapper_with_flatten_observation[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F..::error file=tests/test_environment_compatibility.py,line=203::TestBasicAPI.test_render[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=203::TestBasicAPI.test_render[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=439::test_atari_wrapper_features_and_pixel_preprocessing[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=499::test_native_downscaling_hot_swap[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=499::test_native_downscaling_hot_swap[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=561::test_native_downscaling_grayscale[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=561::test_native_downscaling_grayscale[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F..::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F..::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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.............. [ 49%]
..ssssssssssssssss.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=640::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=688::TestAdvancedWrapperFeatures.test_log_wrapper_tracking[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=688::TestAdvancedWrapperFeatures.test_log_wrapper_tracking[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=725::TestAdvancedWrapperFeatures.test_multi_reward_log_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=725::TestAdvancedWrapperFeatures.test_multi_reward_log_wrapper[tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F..::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F..::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=498::TestJaxTransforms.test_jit_compilation[Pixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F..::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=498::TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F...::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the same pytree structure.%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=498::TestJaxTransforms.test_jit_compilation[NormalizedPixel-tictactoe3d]%0A%0ATypeError: cond branch outputs must have the same pytree structure, but they differ:%0A%0Atrue_fun is render_black at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:572%0Afalse_fun is render_normal at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tictactoe3d.py:575%0A%0Atrue_fun output is a <class 'NoneType'> but false_fun output is a pytree leaf, so their Python types differ.%0A%0ARevise true_fun and/or false_fun so that they have the...*[Comment body truncated]*

</details>

[github-actions]

📁 Previous CI results (run #23743989472)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
.....................sssssssssssssss.............s..................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.... [ 91%]
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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 376.45s (0:06:16) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
..........ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 235.58s (0:03:55) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.................s.. [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7fbc68123ce0>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
........ssssssssssssssss................................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
......s......... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 175.68s (0:02:55) ============

---

This log was automatically created at 2026-03-29 13:34:03 UTC.

[github-actions]

📁 Previous CI results (run #23750101877)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
..........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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................... [ 45%]
.....................sssssssssssssss.............s..................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.... [ 91%]
.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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 359.51s (0:05:59) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 227.16s (0:03:47) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
................s.... [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7f0808033060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..........s....... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 185.19s (0:03:05) ============

---

This log was automatically created at 2026-03-30 12:22:03 UTC.

[github-actions]

📁 Previous CI results (run #23753354008)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssssssssssss....................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........ [ 91%]
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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 363.46s (0:06:03) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss................................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 260.06s (0:04:20) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
...............s.... [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7f8b90223060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))
/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7fdcf142b060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
...........s...... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 172.49s (0:02:52) ============

---

This log was automatically created at 2026-03-30 14:41:23 UTC.

[github-actions]

📁 Previous CI results (run #23759224549)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
.......................ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s [ 91%]
.............                                                            [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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 368.55s (0:06:08) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss................................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 238.72s (0:03:58) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.................s.. [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7f149cf27060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss................................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.........s....... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 170.96s (0:02:50) ============

---

This log was automatically created at 2026-03-30 15:49:54 UTC.

[github-actions]

📁 Previous CI results (run #23759320927)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssssssssssss....................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........ [ 91%]
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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 354.29s (0:05:54) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
..........ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 259.29s (0:04:19) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.

::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
................s.... [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7f55d342f060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
........ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........s........ [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 163.73s (0:02:43) ============

---

This log was automatically created at 2026-03-30 18:00:00 UTC.

[github-actions]

📁 Previous CI results (run #23767668358)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssss...ssssssss..............::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........... [ 91%]
.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 (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[bankheist]
  /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[bankheist]
  /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 [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:521: 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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 357.15s (0:05:57) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss................................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 235.48s (0:03:55) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
................s.... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
............s..... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 170.04s (0:02:50) ============

---

This log was automatically created at 2026-03-30 18:02:58 UTC.

[github-actions]

📁 Previous CI results (run #23793181521)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• testing_tictactoe3d.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssssssssssss....................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........ [ 91%]
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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 363.51s (0:06:03) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
..........ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 242.11s (0:04:02) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
.......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.......s........... [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7f90b8527060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..........s....... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 170.96s (0:02:50) ============

---

This log was automatically created at 2026-03-30 21:15:20 UTC.

[github-actions]

📁 Previous CI results (run #23802437681)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssss...ssssssss..............::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........... [ 91%]
.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 (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[bankheist]
  /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[bankheist]
  /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 [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:521: 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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 362.09s (0:06:02) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.

::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 252.33s (0:04:12) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.................s.. [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7fb281e23060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..........s....... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 172.43s (0:02:52) ============

---

This log was automatically created at 2026-03-31 10:52:55 UTC.

[github-actions]

📁 Previous CI results (run #23807390540)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
.........................ssssssssssssssss......................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
......... [ 91%]
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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 361.36s (0:06:01) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
..........ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
...s............ [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 235.65s (0:03:55) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
................s... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
............s..... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 169.19s (0:02:49) ============

---

This log was automatically created at 2026-03-31 14:35:22 UTC.

[github-actions]

📁 Previous CI results (run #23809526791)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
..........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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................... [ 45%]
.....................sssssssssssssss.............s..................::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/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/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`
.... [ 91%]
.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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 379.65s (0:06:19) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 238.64s (0:03:58) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
.......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..................s. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
............s..... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 179.62s (0:02:59) ============

---

This log was automatically created at 2026-03-31 16:21:34 UTC.

[github-actions]

📁 Previous CI results (run #23818009726)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

...........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssssssssssss....................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........ [ 91%]
s............                                                            [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7f50f0023060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=================================== 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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 357.29s (0:05:57) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
..........ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
...s............ [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 235.69s (0:03:55) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.

::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
................s.... [ 99%]
.                                                                        [100%]/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/_pytest/unraisableexception.py:67: PytestUnraisableExceptionWarning: Exception ignored in: <function _xla_gc_callback at 0x7f520c523060>

Traceback (most recent call last):
  File "/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/lib/__init__.py", line 124, in _xla_gc_callback
    xla_client._xla.collect_garbage()
KeyboardInterrupt

Enable tracemalloc to get traceback where the object was allocated.
See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.
  warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))

=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
.......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.......s........... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 197.94s (0:03:17) ============

---

This log was automatically created at 2026-03-31 17:11:22 UTC.

[github-actions]

📁 Previous CI results (run #23818801538)

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• train_agents.md

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
..........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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................... [ 45%]
.....................sssssssssssssss.............s..................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.... [ 91%]
.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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 377.65s (0:06:17) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
.........ssssssssssssssss................................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
.s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 250.00s (0:04:09) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
.......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
...................s [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
........ssssssssssssssss...............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........s........ [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 184.64s (0:03:04) ============

---

This log was automatically created at 2026-03-31 20:42:05 UTC.

[github-actions]

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

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:

• .gitignore
• docs/source/requirements.txt
• pyproject.toml
• scripts/get_objects_patches.py
• scripts/play.py
• src/jaxatari/core.py
• train_agents.md

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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.................... [ 45%]
............................ssssssssssssssss....................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
........ [ 91%]
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 (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[bankheist]
  /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[bankheist]
  /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 [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[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (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, 139 passed, 17 skipped, 3 warnings in 360.65s (0:06:00) =======

skiing ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s................s............................................... [ 49%]
..........ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s............. [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[skiing]
  /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[skiing]
  /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:676: Game skiing not in list of games with mods
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
=========== 120 passed, 25 skipped, 3 warnings in 234.33s (0:03:54) ============

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..............s...... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

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

tictactoe3d ✅

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

sssssssssssss.::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
sssss.....s............................................... [ 49%]
........ssssssssssssssss................................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
......s......... [ 99%]
.                                                                        [100%]
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tictactoe3d]
  /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[tictactoe3d]
  /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:694: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:483: Game does not have mods registered
SKIPPED [1] tests/test_all_mods.py:542: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tictactoe3d' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tictactoe3d' 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
=========== 109 passed, 36 skipped, 3 warnings in 164.17s (0:02:44) ============

---

This log was automatically created at 2026-03-31 21:03:09 UTC.