Skip to content

Commit 8582a43

Browse files
Update src/ml_flashpoint/core/checkpoint_loader.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent cfa03b0 commit 8582a43

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/ml_flashpoint/core/checkpoint_loader.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ def __init__(
128128
self,
129129
checkpoint_object_manager: CheckpointObjectManager,
130130
replication_manager: ReplicationManager,
131+
*,
131132
global_rank_getter: Callable[[], int] = dist.get_rank,
132133
local_rank_getter: Callable[[], int] = dist.get_node_local_rank,
133134
broadcast_object_list_func: Callable[..., None] = dist.broadcast_object_list,

0 commit comments

Comments
 (0)