nemo_gym.artifacts.rollout_rows

The rollout rows a run writes: the private keys that mark a row, and the row as it is exported.

Module Contents

Name	Description
`_rollout_for_export`	Keep canonical operation evidence in the exporter view, without conversation or capture payloads.
`_strip_capture_payloads`	-
`is_no_generation_placeholder`	Whether `response` carries the `NG_NO_GENERATION_METADATA` marker.

nemo_gym.artifacts.rollout_rows._rollout_for_export(
    result: dict[str, typing.Any]
) -> dict[str, typing.Any]

Keep canonical operation evidence in the exporter view, without conversation or capture payloads.

nemo_gym.artifacts.rollout_rows._strip_capture_payloads(
    result: dict[str, typing.Any]
) -> None

nemo_gym.artifacts.rollout_rows.is_no_generation_placeholder(
    response: collections.abc.Mapping[str, typing.Any]
) -> bool

Whether response carries the NG_NO_GENERATION_METADATA marker.

nemo_gym.artifacts.rollout_rows.NG_DISPATCH_DRAINED_KEY = '_ng_dispatch_drained'

nemo_gym.artifacts.rollout_rows.NG_FAILURE_CLASS_KEY = '_ng_failure_class'

nemo_gym.artifacts.rollout_rows.NG_NO_GENERATION_METADATA = {'ng_no_generation': 'true'}

nemo_gym.artifacts.rollout_rows.NG_NO_PERSIST_KEY = '_ng_no_persist'

nemo_gym.artifacts.rollout_rows._MODEL_CALL_PAYLOAD_KEYS = ('request', 'response', 'request_raw', 'response_raw')