Skip to content

Commit 23eb2d5

Browse files
committed
refactor(checkpoint_consumer): update keep_limit to use BASE_CHECKPOINT_RETENTION_LIMIT for improved clarity
1 parent 336cc5b commit 23eb2d5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

grail/infrastructure/checkpoint_consumer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@
4646
from grail.shared.safetensors_utils import load_model_state_dict
4747

4848
from ..shared.constants import (
49+
BASE_CHECKPOINT_RETENTION_LIMIT,
4950
CHECKPOINT_MILESTONE_INTERVAL,
50-
CHECKPOINT_RETENTION_LIMIT,
5151
DELTA_BASE_INTERVAL,
5252
GRAIL_CHECKPOINT_MOD10,
5353
WINDOW_LENGTH,
@@ -115,7 +115,7 @@ def __init__(
115115
*,
116116
cache_root: Path,
117117
credentials: Any | None,
118-
keep_limit: int = CHECKPOINT_RETENTION_LIMIT,
118+
keep_limit: int = BASE_CHECKPOINT_RETENTION_LIMIT,
119119
) -> None:
120120
self.cache_root = cache_root.expanduser().resolve()
121121
self.cache_root.mkdir(parents=True, exist_ok=True)

0 commit comments

Comments
 (0)