We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4842b9d commit b249973Copy full SHA for b249973
nemo_rl/utils/venvs.py
@@ -102,16 +102,9 @@ def create_local_venv_on_each_node(py_executable: str, venv_name: str):
102
pg = placement_group(bundles=bundles, strategy="STRICT_SPREAD")
103
ray.get(pg.ready())
104
105
- runtime_env = {
106
- "env_vars": {
107
- "VIRTUAL_ENV": "",
108
- "UV_PROJECT_ENVIRONMENT": "",
109
- "NEMO_RL_VENV_DIR": os.environ.get("NEMO_RL_VENV_DIR", "/home/ray/venvs"),
110
- },
111
- }
112
# Launch one actor per node
113
actors = [
114
- _env_builder.options(placement_group=pg, runtime_env=runtime_env).remote(py_executable, venv_name)
+ _env_builder.options(placement_group=pg).remote(py_executable, venv_name)
115
for _ in range(num_nodes)
116
]
117
# ensure setup runs on each node
0 commit comments