Tictactoe3d final submission by imnikitk · Pull Request #269 · k4ntz/JAXAtari

imnikitk · 2026-03-31T20:35:29Z

Hello there,
here is the final version of implemented tictactoe3d from Nikit Kattel and Tenzing sherpa. We needed to change some things in our project as given by the instructor since i have a intel mac which just wouldn't support jax and jaxlib 0.7.0. Here are the changes we had to make: pyproject.toml:

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

github-actions · 2026-03-31T20:48:15Z

📁 Previous CI results (run #23819407224)

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

Base Branch ✅

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

Changed Files ❌

The PR changes files that should not be changed:

scripts/get_objects_patches.py
scripts/play.py
src/jaxatari/core.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.................... [ 45%]
.....................ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
...... [ 91%]
......s......                                                            [100%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/numpy/util.py:277: ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
=========================== short test summary info ============================
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
FAILED tests/test_core_and_wrappers.py::test_native_downscaling_grayscale[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
====== 1 failed, 139 passed, 17 skipped, 3 warnings in 370.61s (0:06:10) =======

skiing ✅

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

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

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

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

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

frostbite ✅

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

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

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

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

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

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

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

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

This log was automatically created at 2026-03-31 20:48:14 UTC.

imnikitk · 2026-03-31T21:11:19Z

Hey there the automatic test for tictactoe3d isn't running here. I don't know why but all the framework tests have passed. Thank you.

github-actions · 2026-03-31T21:17:47Z

📁 Previous CI results (run #23819503511)

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

Base Branch ✅

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

Changed Files ❌

The PR changes files that should not be changed:

scripts/get_objects_patches.py
scripts/play.py
src/jaxatari/core.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

..........................................::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
.........::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.................... [ 45%]
.......................ssssssssssssssss.............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/env_checker.py,line=434::WARN: Not able to test alternative render modes due to the environment not having a spec. Try instantiating the environment through `gymnasium.make`
..s. [ 91%]
.............                                                            [100%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/jax/_src/numpy/util.py:277: ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)
=============================== warnings summary ===============================
tests/test_all_mods.py::test_no_duplicate_mod_keys
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/pygame/pkgdata.py:25: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
    from pkg_resources import resource_stream, resource_exists

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

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

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
=========================== short test summary info ============================
SKIPPED [8] tests/test_environment_compatibility.py:521: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:560: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
FAILED tests/test_core_and_wrappers.py::test_native_downscaling_grayscale[bankheist] - ValueError: Incompatible shapes for broadcasting: (3,) and requested shape (1,)
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
====== 1 failed, 139 passed, 17 skipped, 3 warnings in 362.97s (0:06:02) =======

skiing ✅

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

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

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

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

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

frostbite ✅

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

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

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

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

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

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

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

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

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

github-actions · 2026-03-31T21:19:43Z

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

Base Branch ✅

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

Changed Files ❌

The PR changes files that should not be changed:

scripts/get_objects_patches.py
scripts/play.py
src/jaxatari/core.py

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

Framework Tests ❌

Some framework tests failed. Please check the details below:

bankheist ❌

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

.::error file=tests/test_all_mods.py,line=188::TestModExecution.test_mod_runs_reset_and_10_steps[bankheist-random_spawns-individual]%0A%0AFailed: step() failed at step 0 with mod 'random_spawns': 'BankHeistState' object has no attribute 'current_diagonal_priority'
::error file=tests/test_all_mods.py,line=188::TestModExecution.test_mod_runs_reset_and_10_steps[bankheist-random_spawns-individual]%0A%0AFailed: step() failed at step 0 with mod 'random_spawns': 'BankHeistState' object has no attribute 'current_diagonal_priority'
F......s..::error file=tests/test_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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_all_mods.py,line=809::TestDatatypeConsistency.test_datatype_consistency_across_operations[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[bankheist-random_spawns-individual]%0A%0AFailed: reset() with PixelWrapper+native_downscaling failed for mod 'random_spawns': 'BankHeistState' object has no attribute 'current_diagonal_priority'
::error file=tests/test_all_mods.py,line=396::TestModWithWrappers.test_pixel_native_downscaling_does_not_crash_modded_env[bankheist-random_spawns-individual]%0A%0AFailed: reset() with PixelWrapper+native_downscaling failed for mod 'random_spawns': 'BankHeistState' object has no attribute 'current_diagonal_priority'
F::error file=tests/test_core_and_wrappers.py,line=30::test_pixel_obs_wrapper_with_stacked_frames[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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_all_mods.py,line=605::TestModifications.test_environment_works_after_mods[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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_all_mods.py,line=605::TestModifications.test_environment_works_after_mods[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=203::TestBasicAPI.test_render[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
::error file=tests/test_environment_compatibility.py,line=203::TestBasicAPI.test_render[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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=445::TestWrapperCompatibility.test_wrapper_determinism[NormalizedPixel-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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.ssssssssss::error file=tests/test_environment_compatibility.py,line=498::TestJaxTransforms.test_jit_compilation[LoggedFlattenedPixelAndObject-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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=593::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
Fssssss::error file=tests/test_environment_compatibility.py,line=640::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AFailed: Gymnasium's check_env failed: 'BankHeistState' object has no attribute 'pending_police_scores'
::error file=tests/test_funcenv_adapter.py,line=42::TestGymnasiumApiCompliance.test_gymnasium_env_checker[bankheist]%0A%0AFailed: Gymnasium's check_env failed: 'BankHeistState' object has no attribute 'pending_police_scores'
F::error file=tests/test_environment_compatibility.py,line=944::TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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=49::TestGymnasiumApiCompliance.test_reset_method[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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=49::TestGymnasiumApiCompliance.test_reset_method[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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-bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'%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=70::TestGymnasiumApiCompliance.test_render_method[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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=70::TestGymnasiumApiCompliance.test_render_method[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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=120::TestGymWrapperIntegration.test_preprocessing_wrappers[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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=120::TestGymWrapperIntegration.test_preprocessing_wrappers[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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=130::TestGymWrapperIntegration.test_frame_stack_wrapper[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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=130::TestGymWrapperIntegration.test_frame_stack_wrapper[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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=106::TestGymWrapperIntegration.test_time_limit_wrapper[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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[bankheist]%0A%0AAttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'%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[bankheist]%0A%0AFailed: JIT compilation of the functional transition failed: 'BankHeistState' object has no attribute 'current_diagonal_priority'
::error file=tests/test_funcenv_adapter.py,line=146::TestJaxTransforms.test_jittable_transition[bankheist]%0A%0AFailed: JIT compilation of the functional transition failed: 'BankHeistState' object has no attribute 'current_diagonal_priority'
F.s......... [ 99%]
.                                                                        [100%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/tests/test_all_mods.py:221: Failed: step() failed at step 0 with mod 'random_spawns': 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/tests/test_all_mods.py:425: Failed: reset() with PixelWrapper+native_downscaling failed for mod 'random_spawns': 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1771: AttributeError: 'BankHeistState' object has no attribute 'pending_police_scores'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/games/jax_bankheist.py:1269: AttributeError: 'BankHeistState' object has no attribute 'current_diagonal_priority'
/home/runne...*[Comment body truncated]*

imnikitk closed this Mar 31, 2026

imnikitk reopened this Mar 31, 2026

Merge branch 'dev' into master

045b0a0

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Tictactoe3d final submission #269

Tictactoe3d final submission #269
imnikitk wants to merge 1 commit into
k4ntz:devfrom
imnikitk:master

imnikitk commented Mar 31, 2026

Uh oh!

github-actions Bot commented Mar 31, 2026 •

edited

Loading

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ❌

Uh oh!

imnikitk commented Mar 31, 2026

Uh oh!

github-actions Bot commented Mar 31, 2026 •

edited

Loading

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ❌

Uh oh!

github-actions Bot commented Mar 31, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

Conversation

imnikitk commented Mar 31, 2026

Uh oh!

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

Uh oh!

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ❌

Uh oh!

imnikitk commented Mar 31, 2026

Uh oh!

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

Uh oh!

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ❌

Uh oh!

github-actions Bot commented Mar 31, 2026

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ❌

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

github-actions Bot commented Mar 31, 2026 •

edited

Loading

github-actions Bot commented Mar 31, 2026 •

edited

Loading