Skip to content

Commit 62e4fbc

Browse files
committed
Fix type errors: declare _forked_checkpoint_dir on UnslothService
1 parent 5489fde commit 62e4fbc

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

src/art/local/backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1451,7 +1451,7 @@ async def _experimental_fork_checkpoint(
14511451
print(
14521452
"Invalidated UnslothService _state cache to pick up forked checkpoint"
14531453
)
1454-
service._forked_checkpoint_dir = dest_checkpoint_dir
1454+
service._forked_checkpoint_dir = dest_checkpoint_dir # type: ignore[union-attr]
14551455

14561456
if verbose:
14571457
print(

src/art/unsloth/service.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,7 @@ class UnslothService:
111111
output_dir: str
112112
_is_sleeping: bool = False
113113
_latest_step: int = 0
114+
_forked_checkpoint_dir: str | None = None
114115
_lora_id_counter: int = 1 # Start from 1 since 0 is reserved
115116
# Dedicated mode subprocess state
116117
_vllm_process: subprocess.Popen | None = field(default=None, repr=False) # type: ignore[type-arg]

0 commit comments

Comments
 (0)