From 61dab86350d0376f112fd8e6a07543c86446c058 Mon Sep 17 00:00:00 2001 From: Teodor-Dumitru Ene Date: Tue, 1 Sep 2026 09:23:39 -0500 Subject: [PATCH] fix(ci): Fix L1 on main Signed-off-by: Teodor-Dumitru Ene --- tests/functional/ppo_async_megatron.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/functional/ppo_async_megatron.sh b/tests/functional/ppo_async_megatron.sh index 6d53c48e7bc..4ebaed2a27f 100644 --- a/tests/functional/ppo_async_megatron.sh +++ b/tests/functional/ppo_async_megatron.sh @@ -70,7 +70,7 @@ cd "${PROJECT_ROOT}" grep -q "Separate PPO clusters initialized" "${EXP_DIR}/run1.log" grep -q "Updated generation window: version=0, lead=2, max_age=2" "${EXP_DIR}/run1.log" grep -q "Updated generation window: version=1, lead=1, max_age=2" "${EXP_DIR}/run1.log" -test "$(grep -c "PPO epoch 2/2" "${EXP_DIR}/run1.log")" -eq 2 +test "$(grep -c "Value epoch 2/2" "${EXP_DIR}/run1.log")" -eq 2 test -f "${CKPT_DIR}/step_1/replay_buffer.pt" test -f "${CKPT_DIR}/step_2/replay_buffer.pt" @@ -83,7 +83,7 @@ test -f "${CKPT_DIR}/step_2/replay_buffer.pt" grep -q "Restoring replay buffer from checkpoint" "${EXP_DIR}/run2.log" grep -q "ReplayBuffer restored:" "${EXP_DIR}/run2.log" grep -q "Updated generation window: version=3, lead=1, max_age=1" "${EXP_DIR}/run2.log" -test "$(grep -c "PPO epoch 2/2" "${EXP_DIR}/run2.log")" -eq 2 +test "$(grep -c "Value epoch 2/2" "${EXP_DIR}/run2.log")" -eq 2 test -d "${CKPT_DIR}/step_4/policy/weights" test -d "${CKPT_DIR}/step_4/value/weights"