Skip to content

chore: upgrade Python 3.13.13, sglang 0.5.10, mlflow, pytest, flash-infer#2243

Merged
terrykong merged 4 commits intomainfrom
chore/upgrade-python-3.13.13
Apr 10, 2026
Merged

chore: upgrade Python 3.13.13, sglang 0.5.10, mlflow, pytest, flash-infer#2243
terrykong merged 4 commits intomainfrom
chore/upgrade-python-3.13.13

Conversation

@kajalj22
Copy link
Copy Markdown
Contributor

@kajalj22 kajalj22 commented Apr 9, 2026

Summary

This PR upgrades Python from 3.13.11 to 3.13.13, uv from 0.11.3 to 0.11.6, sglang from 0.5.7.dev0 to the official v0.5.10 release, mlflow from 3.9.0rc0 to 3.11.1, and pytest to 9.0.3 across the NeMo-RL codebase.

Python & uv Upgrade (3.13.11 → 3.13.13, uv 0.11.3 → 0.11.6)

Changes Made

  • Updated .python-version to 3.13.13
  • Updated pyproject.toml requires-python to >=3.13.13
  • Updated Dockerfile to use Python 3.13.13 and uv 0.11.6
  • Updated Dockerfile.ngc_pytorch to use uv 0.11.6
  • Updated pyrefly.toml python-version to 3.13.13
  • Updated .claude/skills/code-style/SKILL.md to Python 3.13.13+
  • Updated research/template_project/ files to Python 3.13.13
  • Regenerated uv.lock with Python 3.13.13

SGLang Upgrade (0.5.7.dev0 → 0.5.10)

Major Changes

  • Upgraded to official release: JustinTong0323 fork v0.5.7.dev0 → official sgl-project/sglang v0.5.10
  • Kernel package renamed: sgl-kernel 0.3.21 → sglang-kernel 0.4.1 (major kernel refactor)
  • Per-extra flashinfer versions: Configured separate flashinfer versions since vllm and sglang extras are mutually exclusive

SGLang v0.5.10 Dependencies Updated

  • flashinfer: 0.6.4 → 0.6.7.post2 (includes bug fixes for NaN issues, memory access, buffer overflows)
  • nvidia-cutlass-dsl: >=4.3.4 → >=4.4.1 (CUTLASS 4.2.1 → 4.4.2)
  • quack-kernels: 0.2.4 → >=0.3.0
  • torchcodec: 0.8.0 → 0.9.1
  • Added dependencies:
    • flash-attn-4>=4.0.0b4
    • watchfiles
    • mistral_common>=1.9.0
    • smg-grpc-servicer>=0.5.0 (replaced separate grpcio packages)

Package Renaming (sgl-kernel → sglang-kernel)

MLflow Upgrade (3.9.0rc0 → 3.11.1)

Changes Made

  • Updated mlflow in core dependencies: >=3.9.0rc0 → >=3.11.1
  • Updated mlflow in override-dependencies: >=3.9.0rc0 → >=3.11.1
  • Updated in uv.lock: v3.10.1 → v3.11.1 (mlflow, mlflow-skinny, mlflow-tracing)

Pytest Override Upgrade (9.0.2 → 9.0.3)

🤖 Generated with Claude Code

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot bot commented Apr 9, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@kajalj22
Copy link
Copy Markdown
Contributor Author

kajalj22 commented Apr 9, 2026

/ok to test fdb6ff1

@kajalj22 kajalj22 added the CI:L1 Run doctests, unit tests, and functional tests label Apr 9, 2026
@kajalj22 kajalj22 requested a review from a team April 10, 2026 00:00
@kajalj22
Copy link
Copy Markdown
Contributor Author

/ok to test b28bd27

@kajalj22
Copy link
Copy Markdown
Contributor Author

/ok to test 7fb7919

@kajalj22
Copy link
Copy Markdown
Contributor Author

/ok to test b28bd27

1 similar comment
@kajalj22
Copy link
Copy Markdown
Contributor Author

/ok to test b28bd27

kajalj22 and others added 4 commits April 10, 2026 11:43
This PR upgrades Python from 3.13.11 to 3.13.13 and uv from 0.11.3 to 0.11.6 across the NeMo-RL codebase.

Changes Made:
- Updated `.python-version` to 3.13.13
- Updated `pyproject.toml` `requires-python` to `>=3.13.13`
- Updated `Dockerfile` to use Python 3.13.13 and uv 0.11.6
- Updated `Dockerfile.ngc_pytorch` to use uv 0.11.6
- Updated `pyrefly.toml` python-version to 3.13.13
- Updated `.claude/skills/code-style/SKILL.md` to Python 3.13.13+
- Updated `research/template_project/` files to Python 3.13.13
- Regenerated `uv.lock` with Python 3.13.13

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Signed-off-by: Kajal Jain <kajalj@nvidia.com>
Upgrade sglang to official v0.5.10 release with separate flashinfer versions for vllm and sglang extras.

Changes Made:
- Upgraded sglang: JustinTong0323 fork v0.5.7.dev0 → official sgl-project/sglang v0.5.10
- Kernel package renamed: sgl-kernel 0.3.21 → sglang-kernel 0.4.1
- Updated sglang dependencies to v0.5.10 requirements:
  * flashinfer: 0.6.4 → 0.6.7.post2
  * nvidia-cutlass-dsl: >=4.3.4 → >=4.4.1
  * quack-kernels: 0.2.4 → >=0.3.0
  * Added: flash-attn-4>=4.0.0b4, watchfiles, mistral_common>=1.9.0
  * Added: smg-grpc-servicer>=0.5.0 (replaced grpcio packages)
  * Updated: torchcodec 0.8.0 → 0.9.1
- Configured per-extra flashinfer versions (mutually exclusive):
  * vllm 0.17.1: flashinfer-python==0.6.4, flashinfer-cubin==0.6.4
  * sglang 0.5.10: flashinfer-python==0.6.7.post2, flashinfer-cubin==0.6.7.post2
- Removed flashinfer from global override-dependencies
- Added flashinfer>=0.5.0 override to relax megatron-core constraint
- Updated global overrides:
  * nvidia-cutlass-dsl: >=4.2.1 → >=4.4.1
  * Kept xgrammar at 0.1.33 (override for sglang's 0.1.32)
- Updated all references: sgl-kernel → sglang-kernel in:
  * [tool.uv.sources]
  * [project.optional-dependencies]
  * [tool.uv] extra-build-variables
  * no-build-isolation-package
  * [tool.uv.extra-build-dependencies]
  * [[tool.uv.dependency-metadata]]
- Regenerated uv.lock with Python 3.13.13

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Signed-off-by: Kajal Jain <kajalj@nvidia.com>
Update mlflow minimum version to 3.11.1 in core dependencies and override-dependencies.

Changes:
- Updated mlflow: >=3.9.0rc0 → >=3.11.1
- Updated in uv.lock: v3.10.1 → v3.11.1 (mlflow, mlflow-skinny, mlflow-tracing)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Signed-off-by: Kajal Jain <kajalj@nvidia.com>
Add override dependency to upgrade pytest to 9.0.3.

Changes:
- Added pytest>=9.0.3 to override-dependencies
- Updated in uv.lock: v9.0.2 → v9.0.3

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Signed-off-by: Kajal Jain <kajalj@nvidia.com>
@kajalj22 kajalj22 force-pushed the chore/upgrade-python-3.13.13 branch from b28bd27 to 83164ef Compare April 10, 2026 15:45
@kajalj22
Copy link
Copy Markdown
Contributor Author

/ok to test 83164ef

@kajalj22 kajalj22 marked this pull request as ready for review April 10, 2026 17:32
@kajalj22 kajalj22 requested review from a team and terrykong as code owners April 10, 2026 17:32
@terrykong terrykong changed the title chore: upgrade Python from 3.13.11 to 3.13.13 chore: upgrade Python from 3.13.11 to 3.13.13, sglang Apr 10, 2026
@terrykong terrykong changed the title chore: upgrade Python from 3.13.11 to 3.13.13, sglang chore: upgrade Python from 3.13.11 to 3.13.13, sglang 0.5.10, mlflow, pytest Apr 10, 2026
@terrykong terrykong changed the title chore: upgrade Python from 3.13.11 to 3.13.13, sglang 0.5.10, mlflow, pytest chore: upgrade Python 3.13.13, sglang 0.5.10, mlflow, pytest, flash-infer Apr 10, 2026
@terrykong terrykong enabled auto-merge (squash) April 10, 2026 18:07
@terrykong terrykong disabled auto-merge April 10, 2026 18:07
@terrykong terrykong enabled auto-merge (squash) April 10, 2026 18:07
@terrykong terrykong merged commit c8d167a into main Apr 10, 2026
34 checks passed
@terrykong terrykong deleted the chore/upgrade-python-3.13.13 branch April 10, 2026 20:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI:L1 Run doctests, unit tests, and functional tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants