Skip to content

Commit b249973

Browse files
committed
fix
Signed-off-by: Hemil Desai <hemild@nvidia.com>
1 parent 4842b9d commit b249973

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

nemo_rl/utils/venvs.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -102,16 +102,9 @@ def create_local_venv_on_each_node(py_executable: str, venv_name: str):
102102
pg = placement_group(bundles=bundles, strategy="STRICT_SPREAD")
103103
ray.get(pg.ready())
104104

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-
}
112105
# Launch one actor per node
113106
actors = [
114-
_env_builder.options(placement_group=pg, runtime_env=runtime_env).remote(py_executable, venv_name)
107+
_env_builder.options(placement_group=pg).remote(py_executable, venv_name)
115108
for _ in range(num_nodes)
116109
]
117110
# ensure setup runs on each node

0 commit comments

Comments
 (0)