@@ -188,39 +188,39 @@ def run(self) -> None:
188188 metrics_list = training_metrics ,
189189 step = self .global_step ,
190190 prefix = "train" ,
191- worker_group_name = self .actor_group_name ,
191+ worker_group_name = self .actor . worker_group_name ,
192192 )
193193 self ._log_ranked_metrics (
194194 metrics_list = actor_time_metrics_per_rank ,
195195 step = self .global_step ,
196196 prefix = "time/actor" ,
197- worker_group_name = self .actor_group_name ,
197+ worker_group_name = self .actor . worker_group_name ,
198198 )
199199 self ._log_ranked_metrics (
200200 metrics_list = rollout_metrics_list ,
201201 step = self .global_step ,
202202 prefix = "rollout" ,
203- worker_group_name = self .actor_group_name ,
203+ worker_group_name = self .actor . worker_group_name ,
204204 )
205205 self ._log_ranked_metrics (
206206 metrics_list = env_time_metrics_per_rank ,
207207 step = self .global_step ,
208208 prefix = "time/env" ,
209- worker_group_name = self .env_group_name ,
209+ worker_group_name = self .env . worker_group_name ,
210210 add_prefix = False ,
211211 )
212212 self ._log_ranked_metrics (
213213 metrics_list = env_metrics_per_rank ,
214214 step = self .global_step ,
215215 prefix = "env" ,
216- worker_group_name = self .env_group_name ,
216+ worker_group_name = self .env . worker_group_name ,
217217 add_prefix = False ,
218218 )
219219 self ._log_ranked_metrics (
220220 metrics_list = rollout_time_metrics_per_rank ,
221221 step = self .global_step ,
222222 prefix = "time/rollout" ,
223- worker_group_name = self .rollout_group_name ,
223+ worker_group_name = self .rollout . worker_group_name ,
224224 add_prefix = False ,
225225 )
226226
0 commit comments