Skip to content

Final Submission: Tutankham#222

Open
JTense wants to merge 148 commits into
k4ntz:devfrom
JTense:master
Open

Final Submission: Tutankham#222
JTense wants to merge 148 commits into
k4ntz:devfrom
JTense:master

Conversation

@JTense

@JTense JTense commented Dec 22, 2025

Copy link
Copy Markdown

This contains a not fully implemented Version of Tutankham.

What functions are currently working:

  1. Move player (UP,DOWN,RIGHT,LEFT) and shoot laser bullets (SPACE + RIGHT/LEFT)
  2. Simple creature movement logic and despawn out of bounds
  3. Load map and background

What functions are currently existing but not in JAX:

  1. Creature spawn logic
  2. Player-Creature-Collisions and Creature-Bullet-Collisions
  3. Not full set of sprites
  4. Laser-flash grenade
  5. Update scores after creature kill
  6. Player respawn

What functions are currently missing:

  1. Animations
  2. Wall-Collisions
  3. Level change
  4. Camera
  5. Full item handling
  6. Creature Pathfinding

@github-actions

github-actions Bot commented Dec 22, 2025

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

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

Base Branch ❌

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

Changed Files ❌

The PR changes files that should not be changed:

  • .gitignore
  • README.md
  • scripts/Tutankham_screenshots/frame_00001.npy
  • scripts/Tutankham_screenshots/frame_00002.npy
  • scripts/Tutankham_screenshots/frame_00003.npy
  • scripts/Tutankham_screenshots/frame_00004.npy
  • scripts/Tutankham_screenshots/frame_00005.npy
  • scripts/Tutankham_screenshots/frame_00006.npy
  • scripts/Tutankham_screenshots/frame_00007.npy
  • scripts/Tutankham_screenshots/frame_00008.npy
  • scripts/Tutankham_screenshots/frame_00009.npy
  • scripts/Tutankham_screenshots/frame_00010.npy

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:

tutankham ❌
::error file=tests/test_environment_compatibility.py,line=209::TestBasicAPI.test_obs_to_flat_array[tutankham]%0A%0ANotImplementedError: Abstract method
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: github-actions-annotate-failures-0.3.0, sugar-1.1.1, syrupy-4.9.1, xdist-3.8.0
created: 2/2 workers
2 workers [126 items]

F::error file=tests/test_core_and_wrappers.py,line=29::test_obs_to_flat_array_with_stacked_observations[tutankham]%0A%0ANotImplementedError: Abstract method%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=42::test_pixel_obs_wrapper_with_stacked_frames[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_core_and_wrappers.py,line=73::test_pixel_and_object_centric_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_core_and_wrappers.py,line=119::test_object_centric_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method%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=142::test_log_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_core_and_wrappers.py,line=195::test_multi_reward_log_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_environment_compatibility.py,line=299::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0AAttributeError: 'TutankhamState' object has no attribute 'astype'
FE::error file=tests/test_core_and_wrappers.py,line=266::test_flatten_observation_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method%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=326::test_log_wrapper_with_flatten_observation[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_environment_compatibility.py,line=299::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%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=357::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: assert False%0A +  where False = isinstance(<jaxatari.spaces.Box object at 0x7fee89945510>, <class 'jaxatari.spaces.Dict'>)%0A +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FE::error file=tests/test_environment_compatibility.py,line=299::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FEE::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=70::TestBasicAPI.test_spaces[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_environment_compatibility.py,line=98::TestBasicAPI.test_reset[tutankham]%0A%0AAttributeError: 'TutankhamState' object has no attribute 'astype'
F::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[Atari-tutankham]%0A%0AAssertionError: Space should be Dict for named tuple observation%0Aassert False%0A +  where False = isinstance(<jaxatari.spaces.Box object at 0x7facf45cffd0>, <class 'jaxatari.spaces.Dict'>)%0A +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FE::error file=tests/test_environment_compatibility.py,line=128::TestBasicAPI.test_step[tutankham]%0A%0AAttributeError: 'TutankhamState' object has no attribute 'astype'
F::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%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.EE::error file=tests/test_environment_compatibility.py,line=186::TestBasicAPI.test_render[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FEE.EE.E::error file=tests/test_environment_compatibility.py,line=489::TestJaxTransforms.test_jit_compilation[Atari-tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE.::error file=tests/test_environment_compatibility.py,line=489::TestJaxTransforms.test_jit_compilation[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FEEE::error file=tests/test_environment_compatibility.py,line=489::TestJaxTransforms.test_jit_compilation[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FEE.EEEssssssssssssssss. [ 57%]
E::error file=tests/test_environment_compatibility.py,line=584::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: Number of observation and space leaves should match%0Aassert 14 == 1%0A +  where 14 = len([Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), Array...), Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), ...])%0A +  and   1 = len([<jaxatari.spaces.Box object at 0x7fee6ce0a3d0>])
F.::error file=tests/test_environment_compatibility.py,line=621::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=669::TestAdvancedWrapperFeatures.test_log_wrapper_tracking[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=704::TestAdvancedWrapperFeatures.test_multi_reward_log_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[Atari-tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE::error file=tests/test_environment_compatibility.py,line=746::TestAdvancedWrapperFeatures.test_atari_wrapper_features[tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%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=437::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE::error file=tests/test_environment_compatibility.py,line=782::TestAdvancedWrapperFeatures.test_log_wrapper_edge_cases[tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%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=919::TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE.EEE.EE.E::error file=tests/test_funcenv_adapter.py,line=79::TestGymnasiumApiCompliance.test_seeding_and_determinism[tutankham]%0A%0ANotImplementedError: Abstract method
F.EEE.EsssssssssE::error file=tests/test_environment_compatibility.py,line=919::TestEdgeCasesAndErrorHandling.test_state_serialization[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE.........                   [100%]
==================================== ERRORS ====================================
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
__ ERROR at setup of TestJaxTransforms.test_jit_compilation[Pixel-tutankham] ___
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[NormalizedPixel-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[LoggedFlattenedPixelAndObject-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[MultiRewardLogged-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymnasiumApiCompliance.test_gymnasium_env_checker[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
__ ERROR at setup of TestGymnasiumApiCompliance.test_reset_method[tutankham] ___
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
___ ERROR at setup of TestGymnasiumApiCompliance.test_step_method[tutankham] ___
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
__ ERROR at setup of TestGymnasiumApiCompliance.test_render_method[tutankham] __
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymWrapperIntegration.test_time_limit_wrapper[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymWrapperIntegration.test_preprocessing_wrappers[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymWrapperIntegration.test_frame_stack_wrapper[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
___ ERROR at setup of TestJaxTransforms.test_jittable_transition[tutankham] ____
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[Pixel-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[PixelAndObjectCentric-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/spaces.py:163: AttributeError: 'TutankhamState' object has no attribute 'astype'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:377: AssertionError: assert False
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/spaces.py:163: AttributeError: 'TutankhamState' object has no attribute 'astype'
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:369: AssertionError: Space should be Dict for named tuple observation
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/spaces.py:163: AttributeError: 'TutankhamState' object has no attribute 'astype'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:218: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:615: AssertionError: Number of observation and space leaves should match
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:218: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:218: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:218: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
=========================== short test summary info ============================
SKIPPED [8] tests/test_environment_compatibility.py:512: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:551: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_regression.py: No snapshot found for game 'tutankham'
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_reset_method[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_render_method[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_preprocessing_wrappers[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_frame_stack_wrapper[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestJaxTransforms::test_jittable_transition[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_obs_to_flat_array[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_obs_to_flat_array_with_stacked_observations[tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_pixel_obs_wrapper_with_stacked_frames[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_pixel_and_object_centric_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_object_centric_wrapper[tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_log_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_multi_reward_log_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Atari-tutankham] - AttributeError: 'TutankhamState' object has no attribute 'astype'
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper[tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_log_wrapper_with_flatten_observation[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper_space_structure[tutankham] - AssertionError: assert False
 +  where False = isinstance(<jaxatari.spaces.Box object at 0x7fee89945510>, <class 'jaxatari.spaces.Dict'>)
 +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_atari_wrapper_features_and_pixel_preprocessing[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_spaces[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_reset[tutankham] - AttributeError: 'TutankhamState' object has no attribute 'astype'
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Atari-tutankham] - AssertionError: Space should be Dict for named tuple observation
assert False
 +  where False = isinstance(<jaxatari.spaces.Box object at 0x7facf45cffd0>, <class 'jaxatari.spaces.Dict'>)
 +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_step[tutankham] - AttributeError: 'TutankhamState' object has no attribute 'astype'
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_render[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Atari-tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_flatten_observation_wrapper_space_structure[tutankham] - AssertionError: Number of observation and space leaves should match
assert 14 == 1
 +  where 14 = len([Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), Array...), Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), ...])
 +  and   1 = len([<jaxatari.spaces.Box object at 0x7fee6ce0a3d0>])
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_normalize_observation_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_tracking[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_multi_reward_log_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Atari-tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_atari_wrapper_features[tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_edge_cases[tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
============ 36 failed, 22 passed, 25 skipped, 43 errors in 21.24s =============


This log was automatically created at 2025-12-22 10:49:56 UTC.

@PaulSeitz PaulSeitz changed the base branch from master to dev December 22, 2025 11:08
@github-actions

github-actions Bot commented Dec 22, 2025

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

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

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
  • README.md
  • scripts/Tutankham_screenshots/frame_00001.npy
  • scripts/Tutankham_screenshots/frame_00002.npy
  • scripts/Tutankham_screenshots/frame_00003.npy
  • scripts/Tutankham_screenshots/frame_00004.npy
  • scripts/Tutankham_screenshots/frame_00005.npy
  • scripts/Tutankham_screenshots/frame_00006.npy
  • scripts/Tutankham_screenshots/frame_00007.npy
  • scripts/Tutankham_screenshots/frame_00008.npy
  • scripts/Tutankham_screenshots/frame_00009.npy
  • scripts/Tutankham_screenshots/frame_00010.npy

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:

tutankham ❌
::error file=tests/test_environment_compatibility.py,line=209::TestBasicAPI.test_obs_to_flat_array[tutankham]%0A%0ANotImplementedError: Abstract method
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: github-actions-annotate-failures-0.3.0, sugar-1.1.1, syrupy-4.9.1, xdist-3.8.0
created: 2/2 workers
2 workers [126 items]

F::error file=tests/test_core_and_wrappers.py,line=29::test_obs_to_flat_array_with_stacked_observations[tutankham]%0A%0ANotImplementedError: Abstract method%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=42::test_pixel_obs_wrapper_with_stacked_frames[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_core_and_wrappers.py,line=73::test_pixel_and_object_centric_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_core_and_wrappers.py,line=119::test_object_centric_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method%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=142::test_log_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_core_and_wrappers.py,line=195::test_multi_reward_log_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_environment_compatibility.py,line=299::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0AAttributeError: 'TutankhamState' object has no attribute 'astype'
F::error file=tests/test_core_and_wrappers.py,line=266::test_flatten_observation_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE::error file=tests/test_core_and_wrappers.py,line=326::test_log_wrapper_with_flatten_observation[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_environment_compatibility.py,line=299::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%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=357::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: assert False%0A +  where False = isinstance(<jaxatari.spaces.Box object at 0x7f16fd901e10>, <class 'jaxatari.spaces.Dict'>)%0A +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FE::error file=tests/test_environment_compatibility.py,line=299::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FEE::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=70::TestBasicAPI.test_spaces[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_environment_compatibility.py,line=98::TestBasicAPI.test_reset[tutankham]%0A%0AAttributeError: 'TutankhamState' object has no attribute 'astype'
F::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[Atari-tutankham]%0A%0AAssertionError: Space should be Dict for named tuple observation%0Aassert False%0A +  where False = isinstance(<jaxatari.spaces.Box object at 0x7f117c85ee90>, <class 'jaxatari.spaces.Dict'>)%0A +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FE::error file=tests/test_environment_compatibility.py,line=128::TestBasicAPI.test_step[tutankham]%0A%0AAttributeError: 'TutankhamState' object has no attribute 'astype'
F::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%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.EE::error file=tests/test_environment_compatibility.py,line=186::TestBasicAPI.test_render[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE.EEE.E::error file=tests/test_environment_compatibility.py,line=489::TestJaxTransforms.test_jit_compilation[Atari-tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE.::error file=tests/test_environment_compatibility.py,line=489::TestJaxTransforms.test_jit_compilation[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FEEE::error file=tests/test_environment_compatibility.py,line=489::TestJaxTransforms.test_jit_compilation[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FEE.EEE.ssssssssssssssss [ 57%]
E.::error file=tests/test_environment_compatibility.py,line=584::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: Number of observation and space leaves should match%0Aassert 14 == 1%0A +  where 14 = len([Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), Array...), Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), ...])%0A +  and   1 = len([<jaxatari.spaces.Box object at 0x7f16e0e747d0>])
FE::error file=tests/test_environment_compatibility.py,line=621::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=669::TestAdvancedWrapperFeatures.test_log_wrapper_tracking[tutankham]%0A%0ANotImplementedError: Abstract method
FE::error file=tests/test_environment_compatibility.py,line=704::TestAdvancedWrapperFeatures.test_multi_reward_log_wrapper[tutankham]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[Atari-tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE::error file=tests/test_environment_compatibility.py,line=746::TestAdvancedWrapperFeatures.test_atari_wrapper_features[tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%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=437::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE::error file=tests/test_environment_compatibility.py,line=782::TestAdvancedWrapperFeatures.test_log_wrapper_edge_cases[tutankham]%0A%0AAttributeError: DynamicJaxprTracer has no attribute items%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=919::TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FE.EEEE.EE.::error file=tests/test_funcenv_adapter.py,line=79::TestGymnasiumApiCompliance.test_seeding_and_determinism[tutankham]%0A%0ANotImplementedError: Abstract method
FE.EEEsssssssss.E::error file=tests/test_environment_compatibility.py,line=919::TestEdgeCasesAndErrorHandling.test_state_serialization[ObjectCentric-tutankham]%0A%0ANotImplementedError: Abstract method%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.E........                   [100%]
==================================== ERRORS ====================================
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
__ ERROR at setup of TestJaxTransforms.test_jit_compilation[Pixel-tutankham] ___
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_observation_spaces[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[NormalizedPixel-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[LoggedFlattenedPixelAndObject-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestJaxTransforms.test_jit_compilation[MultiRewardLogged-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_action_spaces[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[Pixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[NormalizedPixel-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[LoggedFlattenedPixelAndObject-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[MultiRewardLogged-tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymnasiumApiCompliance.test_gymnasium_env_checker[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
__ ERROR at setup of TestGymnasiumApiCompliance.test_reset_method[tutankham] ___
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
___ ERROR at setup of TestGymnasiumApiCompliance.test_step_method[tutankham] ___
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
__ ERROR at setup of TestGymnasiumApiCompliance.test_render_method[tutankham] __
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymWrapperIntegration.test_time_limit_wrapper[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymWrapperIntegration.test_preprocessing_wrappers[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestGymWrapperIntegration.test_frame_stack_wrapper[tutankham] _
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
___ ERROR at setup of TestJaxTransforms.test_jittable_transition[tutankham] ____
[gw1] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[Pixel-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
_ ERROR at setup of TestEdgeCasesAndErrorHandling.test_state_serialization[PixelAndObjectCentric-tutankham] _
[gw0] linux -- Python 3.11.14 /home/runner/work/JAXAtari/JAXAtari/.venv/bin/python
E   NotImplementedError: Abstract method
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/spaces.py:164: AttributeError: 'TutankhamState' object has no attribute 'astype'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:377: AssertionError: assert False
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/spaces.py:164: AttributeError: 'TutankhamState' object has no attribute 'astype'
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:369: AssertionError: Space should be Dict for named tuple observation
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/spaces.py:164: AttributeError: 'TutankhamState' object has no attribute 'astype'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:223: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:615: AssertionError: Number of observation and space leaves should match
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:223: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:223: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:223: AttributeError: DynamicJaxprTracer has no attribute items
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:112: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
=========================== short test summary info ============================
SKIPPED [8] tests/test_environment_compatibility.py:512: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:551: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_regression.py: No snapshot found for game 'tutankham'
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_observation_spaces[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_action_spaces[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[NormalizedPixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[LoggedFlattenedPixelAndObject-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[MultiRewardLogged-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_reset_method[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_render_method[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_preprocessing_wrappers[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_frame_stack_wrapper[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_funcenv_adapter.py::TestJaxTransforms::test_jittable_transition[tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[Pixel-tutankham] - NotImplementedError: Abstract method
ERROR tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[PixelAndObjectCentric-tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_obs_to_flat_array[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_obs_to_flat_array_with_stacked_observations[tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_pixel_obs_wrapper_with_stacked_frames[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_pixel_and_object_centric_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_object_centric_wrapper[tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_log_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_core_and_wrappers.py::test_multi_reward_log_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Atari-tutankham] - AttributeError: 'TutankhamState' object has no attribute 'astype'
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper[tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_log_wrapper_with_flatten_observation[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper_space_structure[tutankham] - AssertionError: assert False
 +  where False = isinstance(<jaxatari.spaces.Box object at 0x7f16fd901e10>, <class 'jaxatari.spaces.Dict'>)
 +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_atari_wrapper_features_and_pixel_preprocessing[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_spaces[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_reset[tutankham] - AttributeError: 'TutankhamState' object has no attribute 'astype'
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Atari-tutankham] - AssertionError: Space should be Dict for named tuple observation
assert False
 +  where False = isinstance(<jaxatari.spaces.Box object at 0x7f117c85ee90>, <class 'jaxatari.spaces.Dict'>)
 +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_step[tutankham] - AttributeError: 'TutankhamState' object has no attribute 'astype'
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_render[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Atari-tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_flatten_observation_wrapper_space_structure[tutankham] - AssertionError: Number of observation and space leaves should match
assert 14 == 1
 +  where 14 = len([Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), Array...), Array([80, 80, 80, 80], dtype=int32, weak_type=True), Array([105, 105, 105, 105], dtype=int32, weak_type=True), ...])
 +  and   1 = len([<jaxatari.spaces.Box object at 0x7f16e0e747d0>])
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_normalize_observation_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_tracking[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_multi_reward_log_wrapper[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[Atari-tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_atari_wrapper_features[tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_log_wrapper_edge_cases[tutankham] - AttributeError: DynamicJaxprTracer has no attribute items
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[FlattenedObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[tutankham] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[ObjectCentric-tutankham] - NotImplementedError: Abstract method
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
============ 36 failed, 22 passed, 25 skipped, 43 errors in 19.72s =============


This log was automatically created at 2025-12-22 11:10:02 UTC.

@IIITyforIII

Copy link
Copy Markdown

Hi, following is the review by the team that is working on Battlezone:

As written in the fork comment by the authors many Features are currently only implemented in "normal" python / not jax compatible. I will only review the jax compatible code (we understood that as the point of the review)

ALE-similarity:
-extracted sprites match the ALE version
-all features work
-not many features are implemented yet (jax compatible)
-only first part of the first map is currently loaded
-in the original ALE version the game remembers directional input (movement works like in pacman)(this is very minor and should be easy to implement)

implementation quality:
-vmap is used correctly in creature step
-current step function has a nice structure but may be harder to extend (for example if 2 of your step functions need to modify 1 of your state variables (maybe collision_step and player_step etc.)) (maybe take a look at _replace if you run into that problem)
-jax.lax.cond is used correctly (although in some places its a bit ugly fe. get_rotation in bullet_step)
-named tuples would be better in some places over arbitrary jnp.arrays(for example bullet state could could be Entity (or some other NamedTuple)
then you can access fe. bullet position from within bullet_step with: bullet_state.position instead of bullet_state[0] -> better for readability and will save you trouble in the future)
if you are worried about vmap, NamedTuple works well with that (as long as each entry of the namedTuple is sth like chex.Array)
-line 568 is doubled (very minor)

@jakobsteinke

jakobsteinke commented Jan 6, 2026

Copy link
Copy Markdown

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

ALE-Similarity: 1.5 / 4

The current implementation differs significantly from the ALE version and does not yet reproduce the core gameplay challenge:

  • Player is not constrained by walls or screen boundaries and can leave the visible area
  • Player–enemy and bullet–enemy collisions are not implemented
  • No clear progression, level change, or fail state
  • Shooting appears limited to the initial phase; later SPACE + LEFT/RIGHT inputs have no effect

Overall, gameplay currently only resembles ALE at a very basic movement level.


Code Quality: 2 / 4 (JAX-Native)

Things that could be improved

  1. Lines 399–407, 420–442: Python if / for inside jitted functions

    • laser_flash_step and spawner_step should use jax.lax.cond and jax.vmap
    • Python control flow breaks JAX compilation
  2. Lines 420, 432, 436: NumPy instead of JAX

    • np.sum()jnp.sum()
    • np.random.random()jax.random with RNG key management
    • In-place ops like ammunition_timer -= 1 won’t work; use jnp.where() or state updates
  3. Lines 71–84: Mixed Python ints and JAX arrays in state

    • Breaks JAX tracing
    • Convert everything to JAX arrays (e.g. player_lives: jnp.ndarray)
  4. Lines 189–227: Action mapping mismatch

    • 18 entries defined for a 9-action space
    • Likely easier to generate dynamically from action_set
  5. Lines 462–480, 533–560: Empty method stubs

    • update_score() and resolve_collisions() do nothing (fine for now, but document it)
    • Line 448 has a duplicate return statement

Performance: 3 / 4

  • Environment runs stably with no crashes, freezes, or major slowdowns
  • Player movement and input handling (arrow keys, shooting) are responsive
  • Step loop executes smoothly and loading times are short
  • Player movement speed may differ slightly from ALE, but does not affect stability
  • Several systems are missing or not fully JAX-native, making scalability unclear once more mechanics are added

@sebiwtt

sebiwtt commented Jan 7, 2026

Copy link
Copy Markdown
Collaborator

Hey there. So I see that the reviews were mostly more on the negative side because the JAX version of your project is still just the bare backbone of the game (and I understood that you first implemented most of the game logic in pure python, right?). That totally okay, we ourselves said that this is fine to do. The downside for you of course now is that you did not really get "real" feedback on what you were working on.

My proposal would be to "integrate" the feedback (where possible) and keep porting your work to JAX, make the tests pass, etc.
When you are at a state that you would like to get some more feedback on, you ping me (or message/email me) and I will take another look.

Keep up the good work!

@github-actions

github-actions Bot commented Mar 24, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ❌

Some framework tests failed. Please check the details below:

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
sssssssss.....ssss.::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=130::test_log_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_core_and_wrappers.py,line=320::test_log_wrapper_with_flatten_observation[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=320::test_log_wrapper_with_flatten_observation[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F [ 49%]
.::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[Atari-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[Pixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[ObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[FlattenedObjectCentric-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but the corresponding output of false_fun has type bool[], so the dtypes do not match.%0A%0ARevise true_fun and/or false_fun so that all output types match.%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[NormalizedPixel-tutankham]%0A%0ATypeError: cond branches must have equal output types but they differ.%0A%0Atrue_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0Afalse_fun is <lambda> at /home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:151%0A%0AThe output of true_fun at path .is_moving has type int8[] but...*[Comment body truncated]*

</details>

@github-actions

github-actions Bot commented Mar 24, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-24 19:37:09 UTC.

@github-actions

github-actions Bot commented Mar 25, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-25 17:54:34 UTC.

@github-actions

github-actions Bot commented Mar 26, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-26 19:39:40 UTC.

@github-actions

github-actions Bot commented Mar 26, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-26 21:10:20 UTC.

@github-actions

github-actions Bot commented Mar 26, 2026

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

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

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:

  • Tutankham_screenshots/night_cover.npy

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:

tutankham ❌
============================= 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
sssssssss.....ssss.::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=145::TestBasicAPI.test_step[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=107::test_object_centric_wrapper[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=181::TestBasicAPI.test_determinism[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=181::TestBasicAPI.test_determinism[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=130::test_log_wrapper[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=226::TestBasicAPI.test_episode_completion[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=351::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[Pixel-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=445::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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%]
::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[Pixel-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[MultiRewardLogged-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[MultiRewardLogged-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
Fssssssssssssssss.::error file=tests/test_environment_compatibility.py,line=593::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=593::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=896::TestEdgeCasesAndErrorHandling.test_state_consistency[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=896::TestEdgeCasesAndErrorHandling.test_state_consistency[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=767::TestAdvancedWrapperFeatures.test_atari_wrapper_features[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=767::TestAdvancedWrapperFeatures.test_atari_wrapper_features[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=918::TestEdgeCasesAndErrorHandling.test_observation_consistency[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=918::TestEdgeCasesAndErrorHandling.test_observation_consistency[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=807::TestAdvancedWrapperFeatures.test_log_wrapper_edge_cases[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=807::TestAdvancedWrapperFeatures.test_log_wrapper_edge_cases[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[Atari-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=853::TestEdgeCasesAndErrorHandling.test_invalid_action_handling[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=853::TestEdgeCasesAndErrorHandling.test_invalid_action_handling[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[Pixel-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[Pixel-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=876::TestEdgeCasesAndErrorHandling.test_extreme_reward_values[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=876::TestEdgeCasesAndErrorHandling.test_extreme_reward_values[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[ObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[PixelAndObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[PixelAndObjectCentric-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[NormalizedPixel-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[NormalizedPixel-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[LoggedFlattenedPixelAndObject-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[MultiRewardLogged-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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=944::TestEdgeCasesAndErrorHandling.test_state_serialization[MultiRewardLogged-tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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_funcenv_adapter.py,line=57::TestGymnasiumApiCompliance.test_step_method[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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_funcenv_adapter.py,line=57::TestGymnasiumApiCompliance.test_step_method[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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_funcenv_adapter.py,line=42::TestGymnasiumApiCompliance.test_gymnasium_env_checker[tutankham]%0A%0AFailed: Gymnasium's check_env failed: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
::error file=tests/test_funcenv_adapter.py,line=42::TestGymnasiumApiCompliance.test_gymnasium_env_checker[tutankham]%0A%0AFailed: Gymnasium's check_env failed: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
F.::error file=tests/test_funcenv_adapter.py,line=106::TestGymWrapperIntegration.test_time_limit_wrapper[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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_funcenv_adapter.py,line=106::TestGymWrapperIntegration.test_time_limit_wrapper[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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_funcenv_adapter.py,line=79::TestGymnasiumApiCompliance.test_seeding_and_determinism[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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_funcenv_adapter.py,line=79::TestGymnasiumApiCompliance.test_seeding_and_determinism[tutankham]%0A%0ATypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given%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_funcenv_adapter.py,line=146::TestJaxTransforms.test_jittable_transition[tutankham]%0A%0AFailed: JIT compilation of the functional transition failed: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
::error file=tests/test_funcenv_adapter.py,line=146::TestJaxTransforms.test_jittable_transition[tutankham]%0A%0AFailed: JIT compilation of the functional transition failed: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
Fs........... [ 99%]
.                                                                        [100%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were given
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_tutankham.py:1316: TypeError: JaxTutankham.resolve_bullet_collisions() takes 3 positional arguments but 4 were...*[Comment body truncated]*

</details>

… and tutankham_mod_plugins

Compability for Whip and Mimic Mod

Add mimic and whip observation and hook

Add new playermovement logic

Resolve laser flash count before adding level

Observation space cleanup
@github-actions

github-actions Bot commented Mar 27, 2026

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

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

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:

  • Tutankham_screenshots/night_cover.npy

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:

tutankham ❌
============================= 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
sssssssss.....ssss.::error file=tests/test_environment_compatibility.py,line=115::TestBasicAPI.test_reset[tutankham]%0A%0AAssertionError: Reset observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=Object(Single, ACTIVE): Pos=(134, 61) | Size=(5, 8) | Ori=3.0 | State=0 | VisID=0, creatur...Object(Single, INACTIVE): Pos=(0, 0) | Size=(0, 0) | Ori=0.0 | State=0 | VisID=0, mod_whip_ready=Array(0, dtype=int32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)).contains
::error file=tests/test_environment_compatibility.py,line=115::TestBasicAPI.test_reset[tutankham]%0A%0AAssertionError: Reset observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=Object(Single, ACTIVE): Pos=(134, 61) | Size=(5, 8) | Ori=3.0 | State=0 | VisID=0, creatur...Object(Single, INACTIVE): Pos=(0, 0) | Size=(0, 0) | Ori=0.0 | State=0 | VisID=0, mod_whip_ready=Array(0, dtype=int32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)).contains
F.::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[tutankham]%0A%0AAssertionError: Step observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=Object(Single, ACTIVE): Pos=(134, 61) | Size=(5, 8) | Ori=3.0 | State=0 | VisID=0, creatur...Object(Single, INACTIVE): Pos=(0, 0) | Size=(0, 0) | Ori=0.0 | State=0 | VisID=0, mod_whip_ready=Array(0, dtype=int32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)).contains
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[tutankham]%0A%0AAssertionError: Step observation should be contained in observation space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=Object(Single, ACTIVE): Pos=(134, 61) | Size=(5, 8) | Ori=3.0 | State=0 | VisID=0, creatur...Object(Single, INACTIVE): Pos=(0, 0) | Size=(0, 0) | Ori=0.0 | State=0 | VisID=0, mod_whip_ready=Array(0, dtype=int32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)).contains
F::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[tutankham]%0A%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[tutankham]%0A%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
F.::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[tutankham]%0A%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[tutankham]%0A%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
F...::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=ObjectGroup(count=4):\n  [ 0] ACTIVE | Pos: (134,  61) | Size: ( 5,  8) | Ori:   3.0 | Stat...Pos: (  0,   0) | Size: ( 0,  0) | Ori:   0.0 | State:  0 | VisID:  0, mod_whip_ready=Array([0, 0, 0, 0], dtype=int32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(4,), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=..., dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(4,), dtype=<class 'jax.numpy.int32'>)).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=ObjectGroup(count=4):\n  [ 0] ACTIVE | Pos: (134,  61) | Size: ( 5,  8) | Ori:   3.0 | Stat...Pos: (  0,   0) | Size: ( 0,  0) | Ori:   0.0 | State:  0 | VisID:  0, mod_whip_ready=Array([0, 0, 0, 0], dtype=int32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(4,), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=..., dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(4,), dtype=<class 'jax.numpy.int32'>)).contains
F::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[tutankham]%0A%0AAssertionError: Expected object shape (4, 142), got (4, 139)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[tutankham]%0A%0AAssertionError: Expected object shape (4, 142), got (4, 139)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
F::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[tutankham]%0A%0AAssertionError: assert 556 == (4 * np.float64(142.0))%0A +  where np.float64(142.0) = get_object_centric_obs_size(Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)))%0A +    where Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)) = observation_space()%0A +      where observation_space = <jax_tutankham.JaxTutankham object at 0x7fc83ff7cd90>.observation_space
::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[tutankham]%0A%0AAssertionError: assert 556 == (4 * np.float64(142.0))%0A +  where np.float64(142.0) = get_object_centric_obs_size(Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)))%0A +    where Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)) = observation_space()%0A +      where observation_space = <jax_tutankham.JaxTutankham object at 0x7fc83ff7cd90>.observation_space
F.::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([[1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e+00, 0.00e+00,\n        0.00e+00, 3.00e+00, 0.00e+00, 0.00e+00, 0....0e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32))%0A +    where contains = Box(low=0.0, high=[[1.6000000e+02 8.0000000e+02 1.6000000e+02 8.0000000e+02 1.0000000e+00\n  2.5500000e+02 2.5500000e+0...000e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([[1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e+00, 0.00e+00,\n        0.00e+00, 3.00e+00, 0.00e+00, 0.00e+00, 0....0e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32))%0A +    where contains = Box(low=0.0, high=[[1.6000000e+02 8.0000000e+02 1.6000000e+02 8.0000000e+02 1.0000000e+00\n  2.5500000e+02 2.5500000e+0...000e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>).contains
F.::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[181,  83,  40],\n         [181,  83,  40],\n         [181,  83,  40],\n         ...,\n         [181,  83,  40],...e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32)))%0A +    where contains = Tuple(Box(low=0, high=255, shape=(4, 210, 160, 3), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[[1.6000000e+02...00e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>)).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[181,  83,  40],\n         [181,  83,  40],\n         [181,  83,  40],\n         ...,\n         [181,  83,  40],...e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32)))%0A +    where contains = Tuple(Box(low=0, high=255, shape=(4, 210, 160, 3), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[[1.6000000e+02...00e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>)).contains
F::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: Observation leaf with shape (4,) not contained in space with shape (16,)%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([1., 1., 1., 1.], dtype=float32))%0A +    where contains = Box(low=0, high=1, shape=(16,), dtype=<class 'jax.numpy.int8'>).contains
::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: Observation leaf with shape (4,) not contained in space with shape (16,)%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([1., 1., 1., 1.], dtype=float32))%0A +    where contains = Box(low=0, high=1, shape=(16,), dtype=<class 'jax.numpy.int8'>).contains
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e+00, 0.00e+00,\n       0.00e+00, 3.00e+00, 0.00e+00, 0.00e+00, 0.00...0e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n       0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00], dtype=float32))%0A +    where contains = Box(low=0.0, high=[1.6000000e+02 8.0000000e+02 1.6000000e+02 8.0000000e+02 1.0000000e+00\n 2.5500000e+02 2.5500000e+02 ...0000000e+00 2.5500000e+02\n 2.5500000e+02 3.6000000e+02 1.0000000e+00], shape=(568,), dtype=<class 'jax.numpy.float32'>).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(Array([1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e+00, 0.00e+00,\n       0.00e+00, 3.00e+00, 0.00e+00, 0.00e+00, 0.00...0e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n       0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00], dtype=float32))%0A +    where contains = Box(low=0.0, high=[1.6000000e+02 8.0000000e+02 1.6000000e+02 8.0000000e+02 1.0000000e+00\n 2.5500000e+02 2.5500000e+02 ...0000000e+00 2.5500000e+02\n 2.5500000e+02 3.6000000e+02 1.0000000e+00], shape=(568,), dtype=<class 'jax.numpy.float32'>).contains
F..::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([181.,  83.,  40., ..., 181.,  83.,  40.], dtype=float32), Array([1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e...e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n       0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00], dtype=float32)))%0A +    where contains = Tuple(Box(low=0, high=255, shape=(403200,), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[1.6000000e+02 8.00000...000000e+00 2.5500000e+02\n 2.5500000e+02 3.6000000e+02 1.0000000e+00], shape=(568,), dtype=<class 'jax.numpy.float32'>)).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([181.,  83.,  40., ..., 181.,  83.,  40.], dtype=float32), Array([1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e...e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n       0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00], dtype=float32)))%0A +    where contains = Tuple(Box(low=0, high=255, shape=(403200,), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[1.6000000e+02 8.00000...000000e+00 2.5500000e+02\n 2.5500000e+02 3.6000000e+02 1.0000000e+00], shape=(568,), dtype=<class 'jax.numpy.float32'>)).contains
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[181,  83,  40],\n         [181,  83,  40],\n         [181,  83,  40],\n         ...,\n         [181,  83,  40],...e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32)))%0A +    where contains = Tuple(Box(low=0, high=255, shape=(4, 210, 160, 3), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[[1.6000000e+02...00e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>)).contains
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-tutankham]%0A%0AAssertionError: Wrapped observation should be contained in wrapper's space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains((Array([[[[181,  83,  40],\n         [181,  83,  40],\n         [181,  83,  40],\n         ...,\n         [181,  83,  40],...e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32)))%0A +    where contains = Tuple(Box(low=0, high=255, shape=(4, 210, 160, 3), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[[1.6000000e+02...00e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>)).contains
F.....::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-tutankham]%0A%0AAssertionError: Observation shape (4, 139) should match space shape (4, 142)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-tutankham]%0A%0AAssertionError: Observation shape (4, 139) should match space shape (4, 142)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
F.::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tutankham]%0A%0AAssertionError: Observation part shape (4, 139) should match space shape (4, 142)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-tutankham]%0A%0AAssertionError: Observation part shape (4, 139) should match space shape (4, 142)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-tutankham]%0A%0AAssertionError: Observation shape (556,) should match space shape (568,)%0Aassert (556,) == (568,)%0A  %0A  At index 0 diff: 556 != 568%0A  %0A  Full diff:%0A    (%0A  -     568,%0A  ?       -%0A  +     556,%0A  ?     +%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-tutankham]%0A%0AAssertionError: Observation shape (556,) should match space shape (568,)%0Aassert (556,) == (568,)%0A  %0A  At index 0 diff: 556 != 568%0A  %0A  Full diff:%0A    (%0A  -     568,%0A  ?       -%0A  +     556,%0A  ?     +%0A    )
F...::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham]%0A%0AAssertionError: Observation part shape (556,) should match space shape (568,)%0Aassert (556,) == (568,)%0A  %0A  At index 0 diff: 556 != 568%0A  %0A  Full diff:%0A    (%0A  -     568,%0A  ?       -%0A  +     556,%0A  ?     +%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham]%0A%0AAssertionError: Observation part shape (556,) should match space shape (568,)%0Aassert (556,) == (568,)%0A  %0A  At index 0 diff: 556 != 568%0A  %0A  Full diff:%0A    (%0A  -     568,%0A  ?       -%0A  +     556,%0A  ?     +%0A    )
F::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tutankham]%0A%0AAssertionError: Observation part shape (4, 139) should match space shape (4, 142)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-tutankham]%0A%0AAssertionError: Observation part shape (4, 139) should match space shape (4, 142)%0Aassert (4, 139) == (4, 142)%0A  %0A  At index 1 diff: 139 != 142%0A  %0A  Full diff:%0A    (%0A        4,%0A  -     142,%0A  ?      ^^%0A  +     139,%0A  ?      ^^%0A    )
F.......... [ 49%]
..........ssssssssssssssss.::error file=tests/test_environment_compatibility.py,line=593::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: Flattened observation should be contained in space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=ObjectGroup(count=4):\n  [ 0] ACTIVE | Pos: (134,  61) | Size: ( 5,  8) | Ori:   3.0 | Stat...  0,   0) | Size: ( 0,  0) | Ori:   0.0 | State:  0 | VisID:  0, mod_whip_ready=Array([0., 0., 0., 0.], dtype=float32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(4,), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=..., dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(4,), dtype=<class 'jax.numpy.int32'>)).contains
::error file=tests/test_environment_compatibility.py,line=593::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[tutankham]%0A%0AAssertionError: Flattened observation should be contained in space%0Aassert Array(False, dtype=bool)%0A +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=ObjectGroup(count=4):\n  [ 0] ACTIVE | Pos: (134,  61) | Size: ( 5,  8) | Ori:   3.0 | Stat...  0,   0) | Size: ( 0,  0) | Ori:   0.0 | State:  0 | VisID:  0, mod_whip_ready=Array([0., 0., 0., 0.], dtype=float32)))%0A +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(4,), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=..., dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(4,), dtype=<class 'jax.numpy.int32'>)).contains
F..::error file=tests/test_environment_compatibility.py,line=640::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[tutankham]%0A%0ATypeError: sub got incompatible shapes for broadcasting: (4, 139), (4, 142).%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[tutankham]%0A%0ATypeError: sub got incompatible shapes for broadcasting: (4, 139), (4, 142).%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.......................::warning file=.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%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:126: AssertionError: Reset observation should be contained in observation space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:165: AssertionError: Step observation should be contained in observation space
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:97: assert (4, 139) == (4, 142)
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:124: assert (4, 139) == (4, 142)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:222: AssertionError: Expected object shape (4, 142), got (4, 139)
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:274: AssertionError: assert 556 == (4 * np.float64(142.0))
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:435: AssertionError: Observation leaf with shape (4,) not contained in space with shape (16,)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:321: AssertionError: Wrapped observation should be contained in wrapper's space
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:372: AssertionError: Observation shape (4, 139) should match space shape (4, 142)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:353: AssertionError: Observation part shape (4, 139) should match space shape (4, 142)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:372: AssertionError: Observation shape (556,) should match space shape (568,)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:353: AssertionError: Observation part shape (556,) should match space shape (568,)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:353: AssertionError: Observation part shape (4, 139) should match space shape (4, 142)
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:622: AssertionError: Flattened observation should be contained in space
/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/numpy/ufuncs.py:1569: TypeError: sub got incompatible shapes for broadcasting: (4, 139), (4, 142).
=============================== 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[tutankham]
  /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[tutankham]
  /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:694: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:542: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:712: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:565: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:732: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:593: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:752: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:605: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:774: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:671: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:787: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [1] tests/test_all_mods.py:809: Game 'tutankham' is not in core.GAME_MODULES
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_reset[tutankham] - AssertionError: Reset observation should be contained in observation space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=Object(Single, ACTIVE): Pos=(134, 61) | Size=(5, 8) | Ori=3.0 | State=0 | VisID=0, creatur...Object(Single, INACTIVE): Pos=(0, 0) | Size=(0, 0) | Ori=0.0 | State=0 | VisID=0, mod_whip_ready=Array(0, dtype=int32)))
 +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)).contains
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_step[tutankham] - AssertionError: Step observation should be contained in observation space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=Object(Single, ACTIVE): Pos=(134, 61) | Size=(5, 8) | Ori=3.0 | State=0 | VisID=0, creatur...Object(Single, INACTIVE): Pos=(0, 0) | Size=(0, 0) | Ori=0.0 | State=0 | VisID=0, mod_whip_ready=Array(0, dtype=int32)))
 +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)).contains
FAILED tests/test_core_and_wrappers.py::test_pixel_and_object_centric_wrapper[tutankham] - assert (4, 139) == (4, 142)
  
  At index 1 diff: 139 != 142
  
  Full diff:
    (
        4,
  -     142,
  ?      ^^
  +     139,
  ?      ^^
    )
FAILED tests/test_core_and_wrappers.py::test_object_centric_wrapper[tutankham] - assert (4, 139) == (4, 142)
  
  At index 1 diff: 139 != 142
  
  Full diff:
    (
        4,
  -     142,
  ?      ^^
  +     139,
  ?      ^^
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Atari-tutankham] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=ObjectGroup(count=4):\n  [ 0] ACTIVE | Pos: (134,  61) | Size: ( 5,  8) | Ori:   3.0 | Stat...Pos: (  0,   0) | Size: ( 0,  0) | Ori:   0.0 | State:  0 | VisID:  0, mod_whip_ready=Array([0, 0, 0, 0], dtype=int32)))
 +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(4,), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=..., dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(4,), dtype=<class 'jax.numpy.int32'>)).contains
FAILED tests/test_core_and_wrappers.py::test_multi_reward_log_wrapper[tutankham] - AssertionError: Expected object shape (4, 142), got (4, 139)
assert (4, 139) == (4, 142)
  
  At index 1 diff: 139 != 142
  
  Full diff:
    (
        4,
  -     142,
  ?      ^^
  +     139,
  ?      ^^
    )
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper[tutankham] - AssertionError: assert 556 == (4 * np.float64(142.0))
 +  where np.float64(142.0) = get_object_centric_obs_size(Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)))
 +    where Dict(player: Dict(x: Box(low=0, high=160, shape=(), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=()...(), dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(), dtype=<class 'jax.numpy.int32'>)) = observation_space()
 +      where observation_space = <jax_tutankham.JaxTutankham object at 0x7fc83ff7cd90>.observation_space
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[ObjectCentric-tutankham] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(Array([[1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e+00, 0.00e+00,\n        0.00e+00, 3.00e+00, 0.00e+00, 0.00e+00, 0....0e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32))
 +    where contains = Box(low=0.0, high=[[1.6000000e+02 8.0000000e+02 1.6000000e+02 8.0000000e+02 1.0000000e+00\n  2.5500000e+02 2.5500000e+0...000e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[PixelAndObjectCentric-tutankham] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains((Array([[[[181,  83,  40],\n         [181,  83,  40],\n         [181,  83,  40],\n         ...,\n         [181,  83,  40],...e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32)))
 +    where contains = Tuple(Box(low=0, high=255, shape=(4, 210, 160, 3), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[[1.6000000e+02...00e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>)).contains
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper_space_structure[tutankham] - AssertionError: Observation leaf with shape (4,) not contained in space with shape (16,)
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(Array([1., 1., 1., 1.], dtype=float32))
 +    where contains = Box(low=0, high=1, shape=(16,), dtype=<class 'jax.numpy.int8'>).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[FlattenedObjectCentric-tutankham] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(Array([1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e+00, 0.00e+00,\n       0.00e+00, 3.00e+00, 0.00e+00, 0.00e+00, 0.00...0e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n       0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00], dtype=float32))
 +    where contains = Box(low=0.0, high=[1.6000000e+02 8.0000000e+02 1.6000000e+02 8.0000000e+02 1.0000000e+00\n 2.5500000e+02 2.5500000e+02 ...0000000e+00 2.5500000e+02\n 2.5500000e+02 3.6000000e+02 1.0000000e+00], shape=(568,), dtype=<class 'jax.numpy.float32'>).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-tutankham] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains((Array([181.,  83.,  40., ..., 181.,  83.,  40.], dtype=float32), Array([1.34e+02, 6.10e+01, 5.00e+00, 8.00e+00, 1.00e...e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n       0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00], dtype=float32)))
 +    where contains = Tuple(Box(low=0, high=255, shape=(403200,), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[1.6000000e+02 8.00000...000000e+00 2.5500000e+02\n 2.5500000e+02 3.6000000e+02 1.0000000e+00], shape=(568,), dtype=<class 'jax.numpy.float32'>)).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[MultiRewardLogged-tutankham] - AssertionError: Wrapped observation should be contained in wrapper's space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains((Array([[[[181,  83,  40],\n         [181,  83,  40],\n         [181,  83,  40],\n         ...,\n         [181,  83,  40],...e+00, 0.00e+00,\n        0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00, 0.00e+00,\n        0.00e+00]], dtype=float32)))
 +    where contains = Tuple(Box(low=0, high=255, shape=(4, 210, 160, 3), dtype=<class 'jax.numpy.uint8'>), Box(low=0.0, high=[[1.6000000e+02...00e+00 2.5500000e+02 2.5500000e+02\n  3.6000000e+02 1.0000000e+00]], shape=(4, 142), dtype=<class 'jax.numpy.float32'>)).contains
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[ObjectCentric-tutankham] - AssertionError: Observation shape (4, 139) should match space shape (4, 142)
assert (4, 139) == (4, 142)
  
  At index 1 diff: 139 != 142
  
  Full diff:
    (
        4,
  -     142,
  ?      ^^
  +     139,
  ?      ^^
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[PixelAndObjectCentric-tutankham] - AssertionError: Observation part shape (4, 139) should match space shape (4, 142)
assert (4, 139) == (4, 142)
  
  At index 1 diff: 139 != 142
  
  Full diff:
    (
        4,
  -     142,
  ?      ^^
  +     139,
  ?      ^^
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[FlattenedObjectCentric-tutankham] - AssertionError: Observation shape (556,) should match space shape (568,)
assert (556,) == (568,)
  
  At index 0 diff: 556 != 568
  
  Full diff:
    (
  -     568,
  ?       -
  +     556,
  ?     +
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[LoggedFlattenedPixelAndObject-tutankham] - AssertionError: Observation part shape (556,) should match space shape (568,)
assert (556,) == (568,)
  
  At index 0 diff: 556 != 568
  
  Full diff:
    (
  -     568,
  ?       -
  +     556,
  ?     +
    )
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[MultiRewardLogged-tutankham] - AssertionError: Observation part shape (4, 139) should match space shape (4, 142)
assert (4, 139) == (4, 142)
  
  At index 1 diff: 139 != 142
  
  Full diff:
    (
        4,
  -     142,
  ?      ^^
  +     139,
  ?      ^^
    )
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_flatten_observation_wrapper_space_structure[tutankham] - AssertionError: Flattened observation should be contained in space
assert Array(False, dtype=bool)
 +  where Array(False, dtype=bool) = contains(TutankhamObservation(player=ObjectGroup(count=4):\n  [ 0] ACTIVE | Pos: (134,  61) | Size: ( 5,  8) | Ori:   3.0 | Stat...  0,   0) | Size: ( 0,  0) | Ori:   0.0 | State:  0 | VisID:  0, mod_whip_ready=Array([0., 0., 0., 0.], dtype=float32)))
 +    where contains = Dict(player: Dict(x: Box(low=0, high=160, shape=(4,), dtype=<class 'jax.numpy.int16'>), y: Box(low=0, high=800, shape=..., dtype=<class 'jax.numpy.float32'>)), mod_whip_ready: Box(low=0, high=1, shape=(4,), dtype=<class 'jax.numpy.int32'>)).contains
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_normalize_observation_wrapper[tutankham] - TypeError: sub got incompatible shapes for broadcasting: (4, 139), (4, 142).
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
====== 20 failed, 89 passed, 36 skipped, 3 warnings in 417.29s (0:06:57) =======


This log was automatically created at 2026-03-27 04:40:22 UTC.

@github-actions

github-actions Bot commented Mar 27, 2026

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

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

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:

  • Tutankham_screenshots/night_cover.npy

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

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-27 04:54:13 UTC.

@github-actions

github-actions Bot commented Mar 27, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-27 04:55:08 UTC.

@github-actions

github-actions Bot commented Mar 28, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-28 12:10:49 UTC.

@github-actions

github-actions Bot commented Mar 28, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-28 14:47:26 UTC.

Added level parameter to collision
@github-actions

github-actions Bot commented Mar 30, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-30 17:47:11 UTC.

@github-actions

github-actions Bot commented Mar 30, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


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

@github-actions

github-actions Bot commented Mar 30, 2026

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-30 19:51:29 UTC.

@JTense JTense changed the title First Submission: Tutankham Final Submission: Tutankham Mar 30, 2026
@github-actions

Copy link
Copy Markdown

Test Report

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

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ✅

All framework tests passed. Good work! 🎉


This log was automatically created at 2026-03-30 21:24:55 UTC.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants