fix pre-commit

Signed-off-by: 0oshowero0 <o0shower0o@outlook.com>

Author: 0oshowero0

tests/test_controller.py

@@ -330,7 +330,9 @@ def test_controller_with_multi_partitions(self, ray_setup):
                 data_fields=data_fields,
             )
         )
-        expected_global_index_2 = torch.tensor(range(part1_index_range, part2_index_range + part1_index_range), dtype=torch.long)
+        expected_global_index_2 = torch.tensor(
+            range(part1_index_range, part2_index_range + part1_index_range), dtype=torch.long
+        )
         assert torch.equal(global_index_2, expected_global_index_2)
         expected_production_status_2 = torch.ones(part2_index_range, len(data_fields), dtype=torch.int8)
         assert torch.equal(production_status_2, expected_production_status_2)

transfer_queue/controller.py

@@ -466,7 +466,7 @@ def mark_consumed(self, task_name: str, global_indices: list[int]):
 
     # ==================== Consumption Status Interface ====================
 
-    def get_consumption_status(self, task_name: str, mask:bool=False) -> tuple[Tensor, Tensor]:
+    def get_consumption_status(self, task_name: str, mask: bool = False) -> tuple[Tensor, Tensor]:
         """
         Get or create consumption status for a specific task.
         Handles dynamic expansion when new samples are added.
@@ -498,7 +498,7 @@ def get_consumption_status(self, task_name: str, mask:bool=False) -> tuple[Tenso
         return partition_global_index, consumption_status
 
     # ==================== Production Status Interface ====================
-    def get_production_status_for_fields(self, field_names: list[str], mask:bool=False) -> tuple[Tensor, Tensor]:
+    def get_production_status_for_fields(self, field_names: list[str], mask: bool = False) -> tuple[Tensor, Tensor]:
         """
         Check if all samples for specified fields are fully produced and ready.