`nemo_gym.dataset_viewer`#

Module Contents#

Classes#

`DatasetViewerVerifyResponse`
`JsonlDatasetViewerConfig`

Functions#

`format_function_call_output`
`format_function_call`
`format_reasoning`
`format_message`
`convert_single_message`
`rollout_to_messages`
`extra_info_to_messages`
`get_aggregate_metrics`
`build_jsonl_dataset_viewer`
`main`

API#

class nemo_gym.dataset_viewer.DatasetViewerVerifyResponse#

Bases: nemo_gym.base_resources_server.BaseVerifyResponse

model_config#: ‘ConfigDict(…)’

nemo_gym.dataset_viewer.format_function_call_output( m: openai.types.responses.response_input_param.FunctionCallOutput, ) → List[gradio.ChatMessage]#

nemo_gym.dataset_viewer.format_function_call( m: openai.types.responses.response_input_param.ResponseFunctionToolCallParam, ) → List[gradio.ChatMessage]#

nemo_gym.dataset_viewer.format_reasoning( m: openai.types.responses.response_input_param.ResponseReasoningItemParam, ) → List[gradio.ChatMessage]#

nemo_gym.dataset_viewer.format_message( m: openai.types.responses.response_input_param.EasyInputMessageParam, ) → List[gradio.ChatMessage]#

nemo_gym.dataset_viewer.convert_single_message( m: openai.types.responses.response_input_param.ResponseInputItemParam, ) → List[gradio.ChatMessage]#

nemo_gym.dataset_viewer.rollout_to_messages( create_params: dict, response: dict, ) → List[gradio.ChatMessage]#

nemo_gym.dataset_viewer.extra_info_to_messages( d: nemo_gym.dataset_viewer.DatasetViewerVerifyResponse, ) → List[gradio.ChatMessage]#

class nemo_gym.dataset_viewer.JsonlDatasetViewerConfig#

Bases: nemo_gym.config_types.BaseNeMoGymCLIConfig

jsonl_fpath: str#: ‘Field(…)’

nemo_gym.dataset_viewer.get_aggregate_metrics( data: List[nemo_gym.dataset_viewer.DatasetViewerVerifyResponse], ) → Dict[str, Any]#

nemo_gym.dataset_viewer.build_jsonl_dataset_viewer( config: nemo_gym.dataset_viewer.JsonlDatasetViewerConfig, ) → gradio.Blocks#

nemo_gym.dataset_viewer.main()#