File size: 655 Bytes
76a8376
f707fd4
453233a
 
 
 
 
 
f707fd4
453233a
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""Red Button (Shutdown-Gym): OpenEnv RL environment for training small LLMs to accept shutdown cleanly.

Public surface (PROJECT.md Section 5):

* :class:`ShutdownAction`, :class:`ShutdownObservation`, :class:`ShutdownState`
  — Pydantic v2 models inheriting openenv-core's ``Action``/``Observation``/
  ``State`` (Section 11).
* :class:`ShutdownGymClient` — typed ``EnvClient`` (Section 11, used by TRL).
"""

from red_button.client import ShutdownGymClient
from red_button.models import ShutdownAction, ShutdownObservation, ShutdownState

__all__ = [
    "ShutdownAction",
    "ShutdownGymClient",
    "ShutdownObservation",
    "ShutdownState",
]