Skip to content

fix(ci_tests): use safe_load in generate ci tests#1763

Open
basheer-cloud wants to merge 4 commits intoNVIDIA-NeMo:mainfrom
basheer-cloud:fix/use-safe-load-in-generate-ci-tests
Open

fix(ci_tests): use safe_load in generate ci tests#1763
basheer-cloud wants to merge 4 commits intoNVIDIA-NeMo:mainfrom
basheer-cloud:fix/use-safe-load-in-generate-ci-tests

Conversation

@basheer-cloud
Copy link
Copy Markdown

I ran into this while reading through the code path around tests/ci_tests/utils/generate_ci_tests.py. Use safe_load in generate ci tests. yaml.load on untrusted input can construct arbitrary Python objects. I kept the patch small and re-ran syntax checks after applying it.

Backward-compat note: This changes a public function signature by replacing mutable defaults with None guards. Call sites stay source-compatible, but callers introspecting defaults will see None now.

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Apr 10, 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.

Comment thread nemo_automodel/components/checkpoint/stateful_wrappers.py Outdated
Comment thread nemo_automodel/components/checkpoint/stateful_wrappers.py Outdated
Comment thread nemo_automodel/components/models/kimi_k25_vl/model.py
Comment thread nemo_automodel/components/models/kimivl/model.py
Comment thread tests/ci_tests/utils/generate_ci_tests.py
@chtruong814 chtruong814 added waiting-for-customer waiting-on-customer Waiting on the original author to respond and removed waiting-for-customer labels Apr 14, 2026
@akoumpa
Copy link
Copy Markdown
Contributor

akoumpa commented Apr 19, 2026

/ok to test d789859

@akoumpa
Copy link
Copy Markdown
Contributor

akoumpa commented Apr 27, 2026

/ok to test e05319e

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

community-request waiting-on-customer Waiting on the original author to respond

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants