Simon ARC Env
January 12, 2024 ยท View on GitHub
| Action Space | Discrete(10) |
| Observation Shape | (32, 32) |
| Observation High | 255 |
| Observation Low | 0 |
| Import | import simon_arc_env gymnasium.make("SimonARC-v0") |
Description
Abstraction and Reasoning Corpus as a Farama Gymnasium environment.
Related project: ARCLE - ARC Learning Environment.
Installation
pip install simon-arc-env
Usage
- Play it by running
python examples/play.py
Press space to draw a pixel.
- Import it to train your RL model
import simon_arc_env
env = gymnasium.make("SimonARC-v0")
The package relies on import side-effects to register the environment
name so, even though the package is never explicitly used, its import is
necessary to access the environment.
Action Space
SimonARC has the action space Discrete(10).
| Value | Key | Meaning |
|---|---|---|
| 0 | TAB | Next page |
| 1 | Return | Submit drawing |
| 2 | Space | Set pixel |
| 3 | Arrow up | Move cursor_y up |
| 4 | Arrow down | Move cursor_y down |
| 5 | Arrow left | Move cursor_x left |
| 6 | Arrow right | Move cursor_x right |
| 7 | m | Increment color |
| 8 | n | Decrement color |
| 9 | s | Adjust canvas size |
| N/A | Escape | Exit game |
| N/A | i | Print info to console |
Observation Space
Observation space Box(low=0, high=255, shape=(32, 32), dtype=np.uint8).
Rewards
If you solve the task you get 10000 in reward.
If you predict the width correct you get 100 in reward.
If you predict the height correct you get 100 in reward.
If your prediction have the correct histogram you get 100 in reward.
Run tests
make test
Version History
- v0: initial version release