Skip to content

Commit afeb5ed

Browse files
committed
fix trinity baseurl support
1 parent 0782173 commit afeb5ed

2 files changed

Lines changed: 1 addition & 4 deletions

File tree

ajet/backbone/main_trinity.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ def patch_runtime_env_to_get_actor():
2525
"""Patch the classmethod of Explorer and Trainer to pass in the runtime env."""
2626
ajet_config = get_ajet_config_from_trinity_side()
2727
runtime_env = get_runtime_env(ajet_config, is_trinity=True)
28+
os.environ.update(runtime_env["env_vars"])
2829

2930
def patched_explorer_get_actor(cls, config: Config):
3031
return (

ajet/backbone/trainer_trinity.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -116,10 +116,6 @@ def __init__(
116116

117117
async def run_async(self):
118118
ajet_config = get_ajet_config_from_trinity_side()
119-
if ajet_config.ajet.enable_experimental_interchange_server:
120-
raise NotImplementedError(
121-
"The experimental reverse proxy is not supported in Trinity backbone yet."
122-
)
123119
warm_up_process(ajet_config)
124120
tracker = await TrinityRolloutManager(
125121
is_eval=self.is_eval,

0 commit comments

Comments
 (0)