Skip to content

Commit c7bb021

Browse files
Supply RNG settings in checkpoint test fixtures
1 parent 0011d8d commit c7bb021

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/megatron/test_checkpoint_symlink.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,8 @@ def _save(self, async_save: bool):
5353
checkpoint_dir = os.path.join(root, 'checkpoint-2')
5454
os.makedirs(checkpoint_dir)
5555
request = FakeAsyncRequest()
56-
args = types.SimpleNamespace(output_dir=root, async_save=async_save)
56+
args = types.SimpleNamespace(
57+
output_dir=root, async_save=async_save, no_save_rng=False, data_parallel_random_init=False)
5758

5859
dist_checkpointing = mock.MagicMock()
5960
dist_checkpointing.save.return_value = request if async_save else None

0 commit comments

Comments
 (0)