State(
    step: 256000000,
    restart: 0,
    next_eval_step: None,
    agent: "<blob:msgpack>",
    value_function: None,
    optimizer: "<blob:msgpack>",
    vf_optimizer: None,
    obs_space: ObsSpace(
        global_features: [],
        entities: {
            "Asteroid": Entity(
                features: [
                    "health",
                    "radius",
                    "x",
                    "y",
                    "dx",
                    "dy",
                ],
            ),
            "Fighter": Entity(
                features: [
                    "x",
                    "y",
                    "dx",
                    "dy",
                    "direction_x",
                    "direction_y",
                    "remaining_time",
                    "gun_cooldown",
                    "player",
                ],
            ),
            "EnemyFighter": Entity(
                features: [
                    "x",
                    "y",
                    "dx",
                    "dy",
                    "direction_x",
                    "direction_y",
                    "gun_cooldown",
                    "player",
                ],
            ),
            "Bullet": Entity(
                features: [
                    "x",
                    "y",
                    "dx",
                    "dy",
                    "lifetime",
                ],
            ),
        },
    ),
    action_space: {
        "FighterAction": CategoricalActionSpace(
            index_to_label: [
                "thrust=On,shoot=On,turn=Left",
                "thrust=On,shoot=On,turn=QuarterLeft",
                "thrust=On,shoot=On,turn=Right",
                "thrust=On,shoot=On,turn=QuarterRight",
                "thrust=On,shoot=On,turn=None",
                "thrust=On,shoot=Off,turn=Left",
                "thrust=On,shoot=Off,turn=QuarterLeft",
                "thrust=On,shoot=Off,turn=Right",
                "thrust=On,shoot=Off,turn=QuarterRight",
                "thrust=On,shoot=Off,turn=None",
                "thrust=Off,shoot=On,turn=Left",
                "thrust=Off,shoot=On,turn=QuarterLeft",
                "thrust=Off,shoot=On,turn=Right",
                "thrust=Off,shoot=On,turn=QuarterRight",
                "thrust=Off,shoot=On,turn=None",
                "thrust=Off,shoot=Off,turn=Left",
                "thrust=Off,shoot=Off,turn=QuarterLeft",
                "thrust=Off,shoot=Off,turn=Right",
                "thrust=Off,shoot=Off,turn=QuarterRight",
                "thrust=Off,shoot=Off,turn=None",
            ],
        ),
    },
)