NVIDIA-NeMo / NVIDIA-NeMo/RL

Custom worker environment doesn't support extra libraries

Open
#2,199 3 comments 0 reactions 0 assignees View on GitHub
bug community-request waiting-on-customer
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

**Describe the bug**

NemoRL allows to create custom environments and custom workers. If any additional package is needed in a custom environment, it's enough to use `py_executable` `uv run --with ` but with custom worker it doesn't work.

So far looks like /opt/venvs/... is created and then reused without considering "--with" package

**Steps/Code to reproduce bug**

TODO (for me): to create a reproducible setup for this bug as it may not be obvious what I mean by custom worker

**Expected behavior**

Either fix `uv run --with` or support `shell=True` in `subprocess.run` to have `py_executable` like `uv add && uv run`

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.