Skip to content

Commit aa96333

Browse files
committed
Fix type: cast Model to TrainableModel for _get_service
1 parent 1d32128 commit aa96333

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/art/local/backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1439,7 +1439,7 @@ async def _experimental_fork_checkpoint(
14391439
# _state is a cached_property that reads get_last_checkpoint_dir() on
14401440
# first access; if it was accessed before the fork, it cached the base
14411441
# model and will never pick up the forked weights.
1442-
service = await self._get_service(model)
1442+
service = await self._get_service(cast(TrainableModel, model))
14431443
if hasattr(service, "_state") and "_state" in service.__dict__:
14441444
del service.__dict__["_state"]
14451445
if verbose:

0 commit comments

Comments
 (0)