Lost Luggage by aswiso · Pull Request #223 · k4ntz/JAXAtari

aswiso · 2025-12-22T11:23:25Z

No description provided.

github-actions · 2025-12-22T11:25:31Z

📁 Previous CI results (run #23410141044)

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

Base Branch ✅

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

Changed Files ❌

The PR changes files that should not be changed:

renovate.json
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:

lost_luggage ❌

::error file=tests/test_core_and_wrappers.py,line=29::test_obs_to_flat_array_with_stacked_observations[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-8.4.2, pluggy-1.6.0
rootdir: /home/runner/work/JAXAtari/JAXAtari
configfile: pyproject.toml
plugins: github-actions-annotate-failures-0.3.0, sugar-1.1.1, syrupy-4.9.1, xdist-3.8.0
created: 2/2 workers
2 workers [126 items]

F::error file=tests/test_environment_compatibility.py,line=209::TestBasicAPI.test_obs_to_flat_array[lost_luggage]%0A%0ANotImplementedError: Abstract method
F.::error file=tests/test_environment_compatibility.py,line=299::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-lost_luggage]%0A%0AAssertionError: Wrapped environment reset observation should not be None%0Aassert None is not None
F.::error file=tests/test_core_and_wrappers.py,line=73::test_pixel_and_object_centric_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=119::test_object_centric_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=299::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=299::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=299::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=195::test_multi_reward_log_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=299::TestWrapperCompatibility.test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=299::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=266::test_flatten_observation_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_core_and_wrappers.py,line=326::test_log_wrapper_with_flatten_observation[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_core_and_wrappers.py,line=357::test_flatten_observation_wrapper_space_structure[lost_luggage]%0A%0AAssertionError: assert False%0A +  where False = isinstance(<jaxatari.spaces.Box object at 0x7fa2645bab90>, <class 'jaxatari.spaces.Dict'>)%0A +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
F::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[Atari-lost_luggage]%0A%0AAssertionError: Observation should not be None%0Aassert None is not None
F.::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=329::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=329::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=329::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=98::TestBasicAPI.test_reset[lost_luggage]%0A%0AAssertionError: Observation should not be None%0Aassert None is not None
F::error file=tests/test_environment_compatibility.py,line=128::TestBasicAPI.test_step[lost_luggage]%0A%0AAssertionError: Step observation should not be None%0Aassert None is not None
F..::error file=tests/test_environment_compatibility.py,line=329::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=329::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[FlattenedObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F......::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[LoggedFlattenedPixelAndObject-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[MultiRewardLogged-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=489::TestJaxTransforms.test_jit_compilation[Atari-lost_luggage]%0A%0AAssertionError: JIT reset observation should not be None%0Aassert None is not None
F........::error file=tests/test_environment_compatibility.py,line=489::TestJaxTransforms.test_jit_compilation[ObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=489::TestJaxTransforms.test_jit_compilation[PixelAndObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=489::TestJaxTransforms.test_jit_compilation[FlattenedObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=489::TestJaxTransforms.test_jit_compilation[LoggedFlattenedPixelAndObject-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=489::TestJaxTransforms.test_jit_compilation[MultiRewardLogged-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
Fssssssssss [ 57%]
ssssss::error file=tests/test_environment_compatibility.py,line=584::TestAdvancedWrapperFeatures.test_flatten_observation_wrapper_space_structure[lost_luggage]%0A%0AAssertionError: Flattened observation should not be None%0Aassert None is not None
F.::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[ObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=437::TestWrapperCompatibility.test_wrapper_determinism[PixelAndObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=828::TestEdgeCasesAndErrorHandling.test_invalid_action_handling[lost_luggage]%0A%0AAssertionError: Valid action should produce valid observation%0Aassert None is not None
F.::error file=tests/test_environment_compatibility.py,line=621::TestAdvancedWrapperFeatures.test_normalize_observation_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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=704::TestAdvancedWrapperFeatures.test_multi_reward_log_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=919::TestEdgeCasesAndErrorHandling.test_state_serialization[ObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=919::TestEdgeCasesAndErrorHandling.test_state_serialization[PixelAndObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=919::TestEdgeCasesAndErrorHandling.test_state_serialization[FlattenedObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::error file=tests/test_environment_compatibility.py,line=919::TestEdgeCasesAndErrorHandling.test_state_serialization[LoggedFlattenedPixelAndObject-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::error file=tests/test_environment_compatibility.py,line=919::TestEdgeCasesAndErrorHandling.test_state_serialization[MultiRewardLogged-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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[lost_luggage]%0A%0AFailed: Gymnasium's check_env failed: Abstract method
F::error file=tests/test_funcenv_adapter.py,line=49::TestGymnasiumApiCompliance.test_reset_method[lost_luggage]%0A%0ANotImplementedError: Abstract method
F.::error file=tests/test_funcenv_adapter.py,line=130::TestGymWrapperIntegration.test_frame_stack_wrapper[lost_luggage]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_funcenv_adapter.py,line=57::TestGymnasiumApiCompliance.test_step_method[lost_luggage]%0A%0ANotImplementedError: Abstract method
F.sssssssss::error file=tests/test_funcenv_adapter.py,line=70::TestGymnasiumApiCompliance.test_render_method[lost_luggage]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_funcenv_adapter.py,line=79::TestGymnasiumApiCompliance.test_seeding_and_determinism[lost_luggage]%0A%0ANotImplementedError: Abstract method
F.::error file=tests/test_funcenv_adapter.py,line=106::TestGymWrapperIntegration.test_time_limit_wrapper[lost_luggage]%0A%0ANotImplementedError: Abstract method
F::error file=tests/test_funcenv_adapter.py,line=120::TestGymWrapperIntegration.test_preprocessing_wrappers[lost_luggage]%0A%0ANotImplementedError: Abstract method
F........                   [100%]
=================================== FAILURES ===================================
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:131: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:308: AssertionError: Wrapped environment reset observation should not be None
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/tests/test_core_and_wrappers.py:377: AssertionError: assert False
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:373: AssertionError: Observation should not be None
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:104: AssertionError: Observation should not be None
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:140: AssertionError: Step observation should not be None
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:499: AssertionError: JIT reset observation should not be None
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:608: AssertionError: Flattened observation should not be None
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/tests/test_environment_compatibility.py:837: AssertionError: Valid action should produce valid observation
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:306: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/wrappers.py:502: ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
/home/runner/work/JAXAtari/JAXAtari/tests/test_funcenv_adapter.py:47: Failed: Gymnasium's check_env failed: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:142: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:142: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:142: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:142: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:142: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:142: NotImplementedError: Abstract method
/home/runner/work/JAXAtari/JAXAtari/src/jaxatari/environment.py:142: NotImplementedError: Abstract method
=========================== short test summary info ============================
SKIPPED [8] tests/test_environment_compatibility.py:512: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_environment_compatibility.py:551: Skipping to debug memory issues in CI
SKIPPED [1] tests/test_funcenv_adapter.py:157: Skipping to debug memory issues in CI
SKIPPED [8] tests/test_regression.py: No snapshot found for game 'lost_luggage'
FAILED tests/test_core_and_wrappers.py::test_obs_to_flat_array_with_stacked_observations[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_obs_to_flat_array[lost_luggage] - NotImplementedError: Abstract method
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[Atari-lost_luggage] - AssertionError: Wrapped environment reset observation should not be None
assert None is not None
FAILED tests/test_core_and_wrappers.py::test_pixel_and_object_centric_wrapper[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_object_centric_wrapper[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[ObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[PixelAndObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[FlattenedObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_multi_reward_log_wrapper[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[LoggedFlattenedPixelAndObject-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapped_reset_and_step[MultiRewardLogged-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_log_wrapper_with_flatten_observation[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_core_and_wrappers.py::test_flatten_observation_wrapper_space_structure[lost_luggage] - AssertionError: assert False
 +  where False = isinstance(<jaxatari.spaces.Box object at 0x7fa2645bab90>, <class 'jaxatari.spaces.Dict'>)
 +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[Atari-lost_luggage] - AssertionError: Observation should not be None
assert None is not None
FAILED tests/test_core_and_wrappers.py::test_atari_wrapper_features_and_pixel_preprocessing[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[ObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[PixelAndObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[FlattenedObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_reset[lost_luggage] - AssertionError: Observation should not be None
assert None is not None
FAILED tests/test_environment_compatibility.py::TestBasicAPI::test_step[lost_luggage] - AssertionError: Step observation should not be None
assert None is not None
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[LoggedFlattenedPixelAndObject-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_observation_shape_and_type[MultiRewardLogged-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[FlattenedObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[LoggedFlattenedPixelAndObject-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[MultiRewardLogged-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[Atari-lost_luggage] - AssertionError: JIT reset observation should not be None
assert None is not None
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[ObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[PixelAndObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[FlattenedObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[LoggedFlattenedPixelAndObject-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestJaxTransforms::test_jit_compilation[MultiRewardLogged-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_flatten_observation_wrapper_space_structure[lost_luggage] - AssertionError: Flattened observation should not be None
assert None is not None
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[ObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestWrapperCompatibility::test_wrapper_determinism[PixelAndObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_invalid_action_handling[lost_luggage] - AssertionError: Valid action should produce valid observation
assert None is not None
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_normalize_observation_wrapper[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestAdvancedWrapperFeatures::test_multi_reward_log_wrapper[lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[ObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[PixelAndObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[FlattenedObjectCentric-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[LoggedFlattenedPixelAndObject-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_environment_compatibility.py::TestEdgeCasesAndErrorHandling::test_state_serialization[MultiRewardLogged-lost_luggage] - ValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}
--------------------
For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
FAILED tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_gymnasium_env_checker[lost_luggage] - Failed: Gymnasium's check_env failed: Abstract method
FAILED tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_reset_method[lost_luggage] - NotImplementedError: Abstract method
FAILED tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_frame_stack_wrapper[lost_luggage] - NotImplementedError: Abstract method
FAILED tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_step_method[lost_luggage] - NotImplementedError: Abstract method
FAILED tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_render_method[lost_luggage] - NotImplementedError: Abstract method
FAILED tests/test_funcenv_adapter.py::TestGymnasiumApiCompliance::test_seeding_and_determinism[lost_luggage] - NotImplementedError: Abstract method
FAILED tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_time_limit_wrapper[lost_luggage] - NotImplementedError: Abstract method
FAILED tests/test_funcenv_adapter.py::TestGymWrapperIntegration::test_preprocessing_wrappers[lost_luggage] - NotImplementedError: Abstract method
================== 51 failed, 50 passed, 25 skipped in 38.83s ==================

This log was automatically created at 2025-12-22 11:25:30 UTC.

FugeFF · 2026-01-06T16:07:08Z

Review of First Submission

Adjusted init function in LostLuggageRenderer before being able to run the game

Modularity: 4/4

Methods are split up well and it´s easy to understand what a specific function does
Variable-naming is also very clear

ALE-similarity: 1.5/4

All sprites and background are drawn properly
Movement is done well and similiar to ALE
Only the bottom counts as a loss. All contact with green space should lead to loss
Cases don´t stop and change sprites when luggage reaches a wall
Unclear how hitboxes work
Suitcases don´t travel from same spawn-point. Sometimes from top, sometimes seemingly random
Phases not implemented - Airplane animation to signify start of new phase missing
Only one piece of luggage at a time in game - Should be multiple
Point-counter isn´t rendered correctly - seems to be stacking the numbers instead of rendering anew

JAX nativeness: 2.5/4

Implementation fully done in JAX
Still missing some functions and classes for observability (f.e. _get_done, LostLuggageObservation)

Performance: 4/4

Through consistent JAX usage, good performance is achieved. The game loads quickly and renders without stutters

sebiwtt · 2026-01-07T06:25:32Z

Quite nice work so far! I think the biggest thing you need to work on, as well said in the review, is the ALE similarity. Really take your time and use the ALE_RAMStateDeltas.py script to play the original version and try to see where your reviewer saw the "weak points" (no insult!).

Then I would also suggest you look at the report from the github actions bot and fix those issues (inconsistent files, tests, etc.)

Keep up the nice work!

LucyLMM · 2026-01-07T12:23:40Z

PKI Review:

ALE-Similarity: X/4
Due to the game not compiling, we can sadly not give any input on how similar it looks and plays
Error Message:
TypeError: LostLuggageRenderer.init() missing 1 required positional argument: 'consts'
Full Error Message attached below
We did try to quick fix, but just jumped from error to error

implementation quality: 3/4
-The code errors when trying to execute (-1)
-A few important functions for observation and scoring are missing (e.g. Observation, get_reward, get_done)
-Otherwise the code seems clear and understandable, jax is being used sensibly

Error_Msg.txt

github-actions · 2026-03-22T19:10:08Z

📁 Previous CI results (run #23510144799)

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

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:

frostbite ✅

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

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

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

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

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

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: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

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

lost_luggage ❌

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

sssssss::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
ss::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
...::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
...::error file=tests/test_core_and_wrappers.py,line=61::test_pixel_and_object_centric_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
s::error file=tests/test_core_and_wrappers.py,line=107::test_object_centric_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=115::TestBasicAPI.test_reset[lost_luggage]%0A%0AAssertionError: Observation should not be None%0Aassert None is not None
::error file=tests/test_environment_compatibility.py,line=115::TestBasicAPI.test_reset[lost_luggage]%0A%0AAssertionError: Observation should not be None%0Aassert None is not None
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[lost_luggage]%0A%0AAssertionError: Step observation should not be None%0Aassert None is not None
::error file=tests/test_environment_compatibility.py,line=145::TestBasicAPI.test_step[lost_luggage]%0A%0AAssertionError: Step observation should not be None%0Aassert None is not None
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
..::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=189::test_multi_reward_log_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=260::test_flatten_observation_wrapper[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=320::test_log_wrapper_with_flatten_observation[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[lost_luggage]%0A%0AAssertionError: assert False%0A +  where False = isinstance(Box(low=0.0, high=1.0, shape=(4,), dtype=<class 'jax.numpy.float32'>), <class 'jaxatari.spaces.Dict'>)%0A +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
::error file=tests/test_core_and_wrappers.py,line=351::test_flatten_observation_wrapper_space_structure[lost_luggage]%0A%0AAssertionError: assert False%0A +  where False = isinstance(Box(low=0.0, high=1.0, shape=(4,), dtype=<class 'jax.numpy.float32'>), <class 'jaxatari.spaces.Dict'>)%0A +    where <class 'jaxatari.spaces.Dict'> = spaces.Dict
F::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-lost_luggage]%0A%0AAssertionError: Wrapped environment reset observation should not be None%0Aassert None is not None
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[Atari-lost_luggage]%0A%0AAssertionError: Wrapped environment reset observation should not be None%0Aassert None is not None
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=439::test_atari_wrapper_features_and_pixel_preprocessing[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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[lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[lost_luggage]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: LostLuggageRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=499::test_native_downscaling_hot_swap[lost_luggage]%0A%0AFailed: Game renderer likely hasn't updated its __init__ to accept 'config'. Error: LostLuggageRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[ObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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[lost_luggage]%0A%0ATypeError: LostLuggageRenderer.__init__() got an unexpected keyword argument 'config'
::error file=tests/test_core_and_wrappers.py,line=561::test_native_downscaling_grayscale[lost_luggage]%0A%0ATypeError: LostLuggageRenderer.__init__() got an unexpected keyword argument 'config'
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[PixelAndObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[FlattenedObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[ObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[PixelAndObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[FlattenedObjectCentric-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=307::TestWrapperCompatibility.test_wrapped_reset_and_step[MultiRewardLogged-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-lost_luggage]%0A%0AAssertionError: Observation should not be None%0Aassert None is not None
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[Atari-lost_luggage]%0A%0AAssertionError: Observation should not be None%0Aassert None is not None
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[LoggedFlattenedPixelAndObject-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::error file=tests/test_environment_compatibility.py,line=337::TestWrapperCompatibility.test_observation_shape_and_type[MultiRewardLogged-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX 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-lost_luggage]%0A%0AValueError: vmap wrapped function must be passed at least one argument containing an array, got empty *args=(None,) and **kwargs={}%0A--------------------%0AFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.
F::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
.::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, potentially causing excessive recompilation. Future versions will require 'flax.struct.PyTreeNode' (and the states/observations/info to flax.struct.dataclass/PyTreeNode). Please refactor your constants class.
::warning file=src/jaxatari/games/jax_lost_luggage.py,line=69::Performance Warning: JaxLostLuggage.consts is a 'NamedTuple'. This prevents JAX from treating constants as static metadata, p...*[Comment body truncated]*

</details>

github-actions · 2026-03-24T20:25:34Z

📁 Previous CI results (run #23511573946)

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

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
scripts/spriteEditor/spriteEditor.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:

frostbite ✅

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

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

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

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

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

lostluggage ❌

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

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

</details>

github-actions · 2026-03-24T21:01:43Z

📁 Previous CI results (run #23512283856)

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

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:

frostbite ✅

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

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

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

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

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 622.53s (0:10:22) ============

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: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [145 items]

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

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: sugar-1.1.1, github-actions-annotate-failures-0.4.0, xdist-3.8.0, syrupy-4.9.1, jaxtyping-0.3.9
created: 2/2 workers
2 workers [157 items]

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

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

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

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

This log was automatically created at 2026-03-24 21:01:41 UTC.

github-actions · 2026-03-24T21:19:42Z

📁 Previous CI results (run #23512529861)

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

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:

frostbite ✅

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

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

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

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

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

lostluggage ❌

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

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

</details>

github-actions · 2026-03-24T21:25:22Z

📁 Previous CI results (run #23513037642)

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

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:

frostbite ✅

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

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

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

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

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

lostluggage ❌

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

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

</details>

github-actions · 2026-03-24T21:38:32Z

📁 Previous CI results (run #23513358486)

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

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:

frostbite ✅

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

ssssss.::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
::warning file=.venv/lib/python3.11/site-packages/pygame/pkgdata.py,line=25::pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
s...............s................................................ [ 49%]
......ssssssssssssssss..............................::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py,line=317::WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
::warning file=.venv/lib/python3.11/site-packages/gymnasium/utils/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[frostbite]
  /home/runner/work/JAXAtari/JAXAtari/.venv/lib/python3.11/site-packages/gymnasium/utils/passive_env_checker.py:317: UserWarning: WARN: No render modes was declared in the environment (env.metadata['render_modes'] is None or not defined), you may have trouble when calling `.render()`.
    logger.warn(

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

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

lostluggage ❌

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

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

</details>

github-actions · 2026-03-24T21:47:00Z

📁 Previous CI results (run #23515019428)

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

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:

frostbite ✅

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

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

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 607.25s (0:10:07) ============

lostluggage ❌

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

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

</details>

github-actions · 2026-03-24T22:22:47Z

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

Base Branch ✅

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

Changed Files ✅

There are no forbidden file changes. Nice 👍

Framework Tests ❌

Some framework tests failed. Please check the details below:

lostluggage ❌

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

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

Added modifications

aswiso · 2026-03-26T14:21:17Z

@sebiwtt, report submitted, I'm close to be done, so some feedback would be greatly appreciated!

Here are the mods I used:

#	Mod Name	Category
1	Linear Movement	Easy
2	Always Zero Score	Easy
3	No Extra Life	Easy
4	Lower Player Upper Bound	Easy
5	More Suitcases	Easy
6	Soft Escape Penalty	Difficult
7	Timed Round / Fast Spawn	Difficult
8	Disappearing Suitcases	Difficult
9	Random Disappearing Suitcases	Difficult

PaulSeitz assigned sebiwtt Jan 6, 2026

aswiso force-pushed the master branch from 67f749e to 56405f0 Compare March 24, 2026 20:49

aswiso changed the title ~~Lost Luggage, Pre-Alpha~~ Lost Luggage Mar 24, 2026

aswiso changed the title ~~Lost Luggage~~ Lost Luggage, Pre-Alpha Mar 24, 2026

aswiso force-pushed the master branch from c5b0917 to cfb7ee0 Compare March 24, 2026 22:20

Readded files

cfb7ee0

aswiso changed the title ~~Lost Luggage, Pre-Alpha~~ Lost Luggage Mar 24, 2026

aswiso force-pushed the master branch from c64dd22 to ed752ae Compare March 25, 2026 16:20

Fixed float to int conflict

ed752ae

Added modifications

Conversation

aswiso commented Dec 22, 2025

Uh oh!

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

Uh oh!

Test Report

Base Branch ✅

Changed Files ❌

Framework Tests ❌

Uh oh!

FugeFF commented Jan 6, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

sebiwtt commented Jan 7, 2026

Uh oh!

LucyLMM commented Jan 7, 2026

Uh oh!

github-actions Bot commented Mar 22, 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 24, 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 24, 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 24, 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 24, 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 24, 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 24, 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 24, 2026

Test Report

Base Branch ✅

Changed Files ✅

Framework Tests ❌

Uh oh!

aswiso commented Mar 26, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

github-actions Bot commented Dec 22, 2025 •

edited

Loading

FugeFF commented Jan 6, 2026 •

edited

Loading

github-actions Bot commented Mar 22, 2026 •

edited

Loading

github-actions Bot commented Mar 24, 2026 •

edited

Loading

github-actions Bot commented Mar 24, 2026 •

edited

Loading

github-actions Bot commented Mar 24, 2026 •

edited

Loading

github-actions Bot commented Mar 24, 2026 •

edited

Loading

github-actions Bot commented Mar 24, 2026 •

edited

Loading

github-actions Bot commented Mar 24, 2026 •

edited

Loading