Skip to content

Add gymnasium-like interface wrappers #25

@joshgreaves

Description

@joshgreaves

We also want to make ARES accessible to people more familiar with gymnasium.

Steps:

  • Fully review the environment interface in gymnasium.
  • Work out the correct mapping between current ARES and gymnasium.
  • Decide how this should be presented in the library.
  • Make the think.

Proposal for representing it in the library:

tasks = load_swebench_env_tasks("some-id")

with environments.SweBenchEnv(...) as dm_env:
    gym_env = environments.wrap_as_gym(dm_env)
    ...

Pros:

  • Encourages uniform use (dm_env version)
  • Enables gym-like for those who want it
  • Reduces number of classes; use the same helper for every type of dm_env.
    Cons:
  • If you use gym-like interface an extra line of boilerplate

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions