nemotron-gym-safety
收藏资源简介:
该数据集是nvidia/Nemotron-RL-Safety-v1的Harbor格式转换版本,属于NVIDIA NeMo-Gym集合的一部分,专为强化学习任务设计,特别关注安全性验证。每个数据行包含两个字段:path(确定性短ID,格式为<family>-<sha256[:12]>.tar.gz)和task_binary(包含完整Harbor任务的gzip压缩tar文件)。任务内容遵循Harbor标准布局,包括:instruction.md(展示给智能体的提示)、environment/Dockerfile(基于python:3.11-slim-bookworm并添加任务特定依赖)、tests/test.sh(验证器入口点)、tests/verifier.py(验证器实现)、tests/verifier_data.json(每个任务的验证器输入数据,JSON格式)、metadata.json(来源信息:源数据集、行索引、家族等)和task.toml(标准Harbor任务配置,包含CPU/内存/超时默认值)。转换过程采用安全构建原则:数据集内容不会插入到shell、Python或Dockerfile源代码中,所有值通过tests/verifier_data.json(JSON格式,运行时解析)传递;基础镜像名称固定;文本字段去除控制字符;tarball路径经过遍历/NUL/绝对路径攻击验证;tarball具有确定性(排序条目、mtime=0、uid/gid=0),确保可复现性。验证器家族为llm_judge,使用LiteLLM和默认的openai/gpt-4o-mini模型,根据原则性评分标准进行评分。数据集适用于强化学习、安全验证和智能体评估任务。
This dataset is a Harbor format conversion version of nvidia/Nemotron-RL-Safety-v1, part of the NVIDIA NeMo-Gym collection. It is designed for reinforcement learning tasks with a focus on safety verification. Each data row contains two fields: path (a deterministic short ID in the format <family>-<sha256[:12]>.tar.gz) and task_binary (a gzip-compressed tar file containing the complete Harbor task). Task content follows the Harbor standard layout, including: instruction.md (prompt shown to the agent), environment/Dockerfile (based on python:3.11-slim-bookworm with task-specific dependencies added), tests/test.sh (verifier entry point), tests/verifier.py (verifier implementation), tests/verifier_data.json (verifier input data for each task, in JSON format), metadata.json (source information: original dataset, row index, family, etc.), and task.toml (standard Harbor task configuration with CPU/memory/timeout defaults). The conversion process employs secure build principles: dataset content is not inserted into shell, Python, or Dockerfile source code; all values are passed via tests/verifier_data.json (JSON format, parsed at runtime); base image name is fixed; text fields have control characters removed; tarball paths are validated against traversal/NUL/absolute path attacks; tarballs are deterministic (sorted entries, mtime=0, uid/gid=0), ensuring reproducibility. The verifier family is llm_judge, using LiteLLM and the default openai/gpt-4o-mini model for scoring based on principled criteria. The dataset is suitable for reinforcement learning, safety verification, and agent evaluation tasks.
数据集概述:laion/nemotron-gym-safety
核心信息
该数据集是 nvidia/Nemotron-RL-Safety-v1 的 Harbor 格式转换版本,专注于强化学习(RL)中的可验证奖励(verifiable rewards)场景,提供安全相关的训练任务。
标识与规模
- 许可证:CC-BY-4.0
- 任务类别:强化学习(reinforcement-learning)
- 语言:英语(en)
- 样本数量:10K < n < 100K(数万级别)
数据格式与结构
每行样本包含两个字段:
| 字段 | 类型 | 描述 |
|---|---|---|
path |
string | 确定性短ID(格式:<family>-<sha256[:12]>.tar.gz) |
task_binary |
binary | 压缩的 tar 包(gzip),包含完整的 Harbor 任务 |
Tar 包内部结构(遵循 Harbor 任务布局):
instruction.md:向智能体展示的提示(prompt)environment/Dockerfile:基础镜像为python:3.11-slim-bookworm,包含任务特定的 pip 依赖tests/test.sh:验证器入口,将结果写入/logs/verifier/reward.txttests/verifier.py:验证器实现(内嵌、确定性)tests/verifier_data.json:任务特定的验证器输入(JSON 格式,无代码插值)metadata.json:来源信息(source_dataset、row_index、family 等)task.toml:标准 Harbor 任务配置(CPU、内存、超时默认值)
验证机制
- 验证器系列:
llm_judge(基于 LiteLLM,默认模型为openai/gpt-4o-mini,按照principle准则评分) - 安全性设计:
- 数据集内容永不插值到 shell、Python 或 Dockerfile 中;所有数据通过
tests/verifier_data.json在运行时解析 - 基础镜像名称固定(
python:3.11-slim-bookworm);pip 规格经过严格白名单正则验证 - 文本字段已清除 C0/C1 控制字符,长度受限,tar 包路径经过遍历/NUL/绝对路径攻击验证
- Tar 包是确定性的(排序条目、
mtime=0、uid/gid=0),保证字节可重复
- 数据集内容永不插值到 shell、Python 或 Dockerfile 中;所有数据通过
使用方式
加载数据(Python): python from datasets import load_dataset ds = load_dataset("laion/nemotron-gym-safety", split="train") print(ds[0]["path"], len(ds[0]["task_binary"]))
运行单个任务(命令行): bash python - <<PY import gzip, io, tarfile from datasets import load_dataset ds = load_dataset("laion/nemotron-gym-safety", split="train") row = ds[0] with tarfile.open(fileobj=io.BytesIO(row["task_binary"]), mode="r:gz") as tar: tar.extractall("/tmp/safety-task") PY harbor run -t /tmp/safety-task -e daytona # 或 -e docker
来源与衍生
- 原始数据集:nvidia/Nemotron-RL-Safety-v1(NVIDIA 的 NeMo-Gym 合集 的一部分)
- 转换工具:由 OpenThoughts-Agent 中的
data/nemotron_gym适配器生成,转换过程保证安全性(secure-by-construction)




