update val_metrics to include target_dataset_name after logging

Author: binary-husky

ajet/backbone/trainer_verl.py

@@ -1002,7 +1002,6 @@ def _rollout_val_dataset(self, target_dataset, target_dataset_name, mode, epoch)
 
         val_metrics = {
             "global_steps": self.global_steps,
-            "target_dataset_name": target_dataset_name,
             "pass_n": pass_n,
             "total_tasks": len(task_results),
             "num_all_success_tasks": num_all_success_tasks,
@@ -1031,6 +1030,7 @@ def _rollout_val_dataset(self, target_dataset, target_dataset_name, mode, epoch)
         )
 
         self.verl_logger.log(data=val_metrics, step=self.global_steps)
+        val_metrics.update({"target_dataset_name": target_dataset_name})
 
         return ctx_trackers, tasks, val_metrics