fix codecheck

Author: liuximeng

recipe/transfer_queue/agent_loop.py

@@ -69,8 +69,5 @@ def _performance_metrics(self, metrics: list[list[dict[str, str]]], output: Data
 
     def create_transferqueue_client(self, controller_info, config):
         ray.get(
-            [
-                worker.create_transferqueue_client.remote(controller_info, config)
-                for worker in self.agent_loop_workers
-            ]
+            [worker.create_transferqueue_client.remote(controller_info, config) for worker in self.agent_loop_workers]
         )

recipe/transfer_queue/ray_trainer.py

@@ -41,8 +41,8 @@
 from tqdm import tqdm
 from transfer_queue import (
     BatchMeta,
-    TransferQueueController,
     SimpleStorageUnit,
+    TransferQueueController,
     get_placement_group,
     process_zmq_server_info,
 )
@@ -416,12 +416,14 @@ def __init__(
     def _initialize_data_system(self):
         # 1. initialize TransferQueueStorage
         train_data_size = (
-                self.config.data.train_batch_size * self.config.trainer.num_global_batch *
-                self.config.actor_rollout_ref.rollout.n
+            self.config.data.train_batch_size
+            * self.config.trainer.num_global_batch
+            * self.config.actor_rollout_ref.rollout.n
         )
         val_data_size = (
-                self.val_batch_size * self.config.trainer.num_global_batch *
-                self.config.actor_rollout_ref.rollout.val_kwargs.n
+            self.val_batch_size
+            * self.config.trainer.num_global_batch
+            * self.config.actor_rollout_ref.rollout.val_kwargs.n
         )
 
         total_storage_size = train_data_size + val_data_size
@@ -667,9 +669,7 @@ def _validate(self):
             if self.config.reward_model.enable and test_batch[0]["reward_model"]["style"] == "model":
                 return {}
 
-            asyncio.run(
-                self.data_system_client.async_put(data=test_batch, partition_id=f"val_{self.global_steps - 1}")
-            )
+            asyncio.run(self.data_system_client.async_put(data=test_batch, partition_id=f"val_{self.global_steps - 1}"))
 
             # Store original inputs
             batch_meta = asyncio.run(
@@ -958,9 +958,7 @@ def init_workers(self):
                 config=self.config, worker_group=self.actor_rollout_wg, rm_wg=self.rm_wg
             )
 
-            self.async_rollout_manager.create_transferqueue_client(
-                self.data_system_controller_info, self.config
-            )
+            self.async_rollout_manager.create_transferqueue_client(self.data_system_controller_info, self.config)
 
     def _save_checkpoint(self):
         from verl.utils.fs import local_mkdir_safe