Skip to content

Commit d35318f

Browse files
author
kip-cxj
committed
fix pre-commit
1 parent 9db2994 commit d35318f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/checkpoint_engine/test_kimi_checkpoint_engine.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,12 @@ def test_kimi_checkpoint_engine(
6262
trainer = create_trainer_worker_group(model_path, trainer_pool, "kimi_ckpt_engine", checkpoint_kwargs)
6363
trainer.reset()
6464
rollout = create_rollout_worker_group(
65-
model_path, rollout_pool, "kimi_ckpt_engine", checkpoint_kwargs, device=get_device_name(), check_allclose=check_allclose
65+
model_path,
66+
rollout_pool,
67+
"kimi_ckpt_engine",
68+
checkpoint_kwargs,
69+
device=get_device_name(),
70+
check_allclose=check_allclose,
6671
)
6772

6873
world_size = trainer.world_size + rollout.world_size

0 commit comments

Comments
 (0)