Skip to content

Keep non-JPEG channel shapes in raw storage - #356

Open
betacatsling wants to merge 1 commit into
Farama-Foundation:mainfrom
betacatsling:fix/jpeg-compatible-image-shapes
Open

Keep non-JPEG channel shapes in raw storage#356
betacatsling wants to merge 1 commit into
Farama-Foundation:mainfrom
betacatsling:fix/jpeg-compatible-image-shapes

Conversation

@betacatsling

Copy link
Copy Markdown

The image-space predicate treats every uint8 HWC channel count as JPEG-compatible. Collection therefore fails for singleton-channel, two-channel and RGBA observations. Restrict automatic JPEG encoding to two-dimensional grayscale and three-channel RGB; store other shapes as raw arrays so all channels and values survive.

Validation: 9 failed / 2 passed on the unchanged base; 130 passed after the fix, including real Arrow, Parquet and HDF5 round trips. Black/isort and diff checks passed. CPU/local storage only; no GPU, MuJoCo or remote dataset claims. The relevant API documentation is updated.

AI assistance was used for implementation and local verification.

Fixes #355.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Default JPEG encoding rejects valid non-RGB observation shapes

1 participant