Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 0 additions & 14 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -166,20 +166,6 @@ Aggregate metrics: results/mcqa_rollouts_aggregate_metrics.json

For per-task pass rates, see the [`gym eval profile`](https://docs.nvidia.com/nemo/gym/main/reference/cli-commands) command.

### Using the NeMo-Gym Container with VLM or Audio/Video Benchmarks

The NeMo-Gym container omits packages with bundled codec libraries
(`opencv-python-headless`, `torchvision`, `torchaudio`) to avoid shipping
royalty-bearing binaries. If you are running VLM or audio/video benchmarks
inside the container, restore them first:

```bash
bash docker/install_codec_deps.sh
```

This installs the packages at the same versions used during the container
build. It is safe to run multiple times.

### Next Steps

- **[Browse Environments](#-available-environments)** — Browse available environments for evaluation and training.
Expand Down
25 changes: 0 additions & 25 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -24,14 +24,12 @@ FROM ${BASE_IMAGE} AS base
ENV NEMO_GYM_CONTAINER=1
USER root

ARG ENROOT_VERSION=3.5.0
RUN <<"EOF" bash -exu -o pipefail
export DEBIAN_FRONTEND=noninteractive
export TZ=America/Los_Angeles

apt-get update
apt-get install -y --no-install-recommends \
bash \
jq \
curl \
git \
Expand All @@ -40,29 +38,6 @@ apt-get install -y --no-install-recommends \
less \
vim

apt-get purge -y \
ffmpeg \
libavcodec* \
libavformat* \
libavutil* \
libswscale* \
libswresample* \
libx264* \
libx265* \
libfdk-aac* \
libmp3lame* \
2>/dev/null || true

apt-get autoremove -y

# Install enroot for the EnrootProvider sandbox backend.
# GitHub releases provide signed .debs for amd64 and arm64.
arch=$(dpkg --print-architecture)
curl -fSsL -o /tmp/enroot.deb \
"https://github.com/NVIDIA/enroot/releases/download/v${ENROOT_VERSION}/enroot_${ENROOT_VERSION}-1_${arch}.deb"
apt-get install -y /tmp/enroot.deb
rm /tmp/enroot.deb

apt-get clean
rm -rf /var/lib/apt/lists/*
EOF
Expand Down
39 changes: 0 additions & 39 deletions docker/install_codec_deps.sh

This file was deleted.

26 changes: 2 additions & 24 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -314,17 +314,6 @@ dev = [
"requests-mock",
]

[dependency-groups]
# Codec-bearing packages excluded from the shipped package and container.
# Not published to PyPI. Versions are resolved and tracked in uv.lock.
# docker/install_codec_deps.sh pins these explicitly to their uv.lock versions;
# update the script when uv lock bumps them.
codec = [
"opencv-python-headless",
"torchvision",
"torchaudio",
]

[build-system]
requires = ["setuptools>=61", "setuptools-scm"]
build-backend = "setuptools.build_meta"
Expand All @@ -344,21 +333,10 @@ constraint-dependencies = [
override-dependencies = [
# segale pins spacy==3.8.4 which has no cp313 wheels; 3.8.13+ does.
"spacy>=3.8.13",

# Codec-bearing packages — never installed in the shipped package or container.
# Binary wheels bundle libavcodec / static FFmpeg executables (H.264, H.265, AAC)
# that may incur MPEG-LA royalties. Servers that genuinely need these declare
# them in their own overrides.txt (requirements.txt servers) or
# [tool.uv] override-dependencies (pyproject.toml servers), which takes
# precedence. Reinstall for testing via docker/install_codec_deps.sh.
"opencv-python-headless; sys_platform == 'never'",
"torchvision; sys_platform == 'never'",
"torchaudio; sys_platform == 'never'",
]
# NOTE: server venvs install nemo-gym editably and inherit this list, so anything
# here is also silently dropped from them (uv >=0.11.24). Servers that need a
# dropped package must re-enable it via their own overrides.txt or
# [tool.uv] override-dependencies. Context: https://github.com/NVIDIA-NeMo/Gym/pull/1835
# here is also silently dropped from them (uv >=0.11.24). Don't add a package any
# server needs. Context: https://github.com/NVIDIA-NeMo/Gym/pull/1835
exclude-dependencies = [
# fasttext-wheel has no cp313 wheels and fails to compile on Python 3.13
# (missing #include <cstdint> in src/args.cc). Excluded as a transitive dep
Expand Down
1 change: 0 additions & 1 deletion resources_servers/longmt_eval/overrides.txt

This file was deleted.

9 changes: 0 additions & 9 deletions resources_servers/vlm_eval_kit/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -102,14 +102,5 @@ requires = ["setuptools>=61", "setuptools-scm"]
where = [".."]
include = ["local_vllm_model"]

[tool.uv]
# Re-enable codec-bearing packages blocked by the root nemo-gym override.
# vlm_eval_kit requires these for vision model evaluation.
override-dependencies = [
"opencv-python>=4.7.0.72",
"decord2>=3.0.0",
"torchvision",
]

[tool.uv.sources]
nemo-gym = { path = "../..", editable = true }
1 change: 0 additions & 1 deletion resources_servers/wmt_translation/overrides.txt

This file was deleted.

Loading
Loading