fix: add bucket_size property with getter and setter

Signed-off-by: jianjunzhong <jianjunzhong@foxmail.com>

Author: jianjunzhong

tests/checkpoint_engine/test_naive_correctness.py

@@ -43,7 +43,7 @@ def init_config() -> DictConfig:
     config.trainer.nnodes = 1
     config.actor_rollout_ref.actor.use_dynamic_bsz = True
     config.actor_rollout_ref.model.path = os.path.expanduser("~/models/Qwen/Qwen3-VL-2B-Instruct")
-    config.actor_rollout_ref.rollout.name = os.environ["ROLLOUT_NAME"]
+    config.actor_rollout_ref.rollout.name = os.environ.get("ROLLOUT_NAME", "vllm")
     config.actor_rollout_ref.rollout.skip_tokenizer_init = False
     config.actor_rollout_ref.rollout.max_num_seqs = 256
     config.actor_rollout_ref.rollout.gpu_memory_utilization = 0.8

verl/checkpoint_engine/base.py

@@ -731,6 +731,16 @@ def __init__(self, bucket_size: int, is_master: bool = False) -> None:
         self.bucket_size = bucket_size
         self.is_master = is_master
 
+    @property
+    def bucket_size(self) -> int:
+        """Return the bucket size in bytes."""
+        return self._bucket_size
+
+    @bucket_size.setter
+    def bucket_size(self, value: int):
+        """Set the bucket size in bytes."""
+        self._bucket_size = value
+
     def prepare(self):
         raise NotImplementedError
 

verl/checkpoint_engine/kimi_checkpoint_engine.py

@@ -245,6 +245,16 @@ def __init__(
         self.initialized = False
         self.checkpoint_name = "kimi_checkpoint_engine"
 
+    @property
+    def bucket_size(self) -> int:
+        """Return the bucket size in bytes."""
+        return self._bucket_size
+
+    @bucket_size.setter
+    def bucket_size(self, value: int):
+        """Set the bucket size in bytes."""
+        self._bucket_size = value
+
     def prepare(self) -> MasterMetadata:
         if self.is_master:
             self.ip = ray.util.get_node_ip_address().strip("[]")