Skip to content

Commit 801ddfc

Browse files
committed
test(common): make EMA schema fixture valid
1 parent 179d9d7 commit 801ddfc

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

source/tests/common/test_argcheck_training.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,9 @@
99
def test_ema_checkpoint_retention_is_left_for_runtime_inheritance() -> None:
1010
training_argument = training_args()
1111

12-
normalized = training_argument.normalize_value({"max_ckpt_keep": 7})
12+
normalized = training_argument.normalize_value(
13+
{"numb_steps": 1, "max_ckpt_keep": 7}
14+
)
1315
training_argument.check_value(normalized, strict=True)
1416

1517
assert normalized["ema_ckpt_keep"] is None

0 commit comments

Comments
 (0)