Skip to content

Commit baf5cff

Browse files
committed
ci: Fix nightly test prorlvs qwen, increase ci metrics table width
Signed-off-by: Kajal Jain <kajalj@nvidia.com>
1 parent 39e5415 commit baf5cff

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

tests/check_metrics.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,8 +211,7 @@ def main():
211211
# Initialize rich console
212212
console = Console()
213213

214-
# Create a table
215-
table = Table(title="Metric Checks", min_width=150, width=args.table_width)
214+
table = Table(title="Metric Checks", min_width=220, width=args.table_width)
216215
table.add_column("Status", style="bold")
217216
table.add_column("Check", style="dim")
218217
table.add_column("Value", style="cyan")

tests/test_suites/llm/prorlv2-qwen2.5-math-1.5b-instruct-1n8g-fsdp2tp1.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ exit_if_max_steps_reached
1414

1515
# Run the experiment
1616
cd $PROJECT_ROOT
17-
uv run examples/run_grpo_math.py \
17+
uv run examples/run_grpo.py \
1818
--config $CONFIG_PATH \
1919
grpo.max_num_steps=$MAX_STEPS \
2020
logger.log_dir=$LOG_DIR \

tests/test_suites/nightly_gb200.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@ tests/test_suites/llm/grpo-qwen2.5-7b-instruct-4n4g-fsdp2tp2.v3.sh
1414
tests/test_suites/llm/grpo-llama3.2-1b-instruct-1n4g-megatron.sh
1515
tests/test_suites/llm/grpo-llama3.2-1b-instruct-1n4g-megatron_generation.sh
1616

17-
# Functional moonlight run
18-
tests/test_suites/llm/grpo-moonlight-16ba3b-4n4g-megatron.sh
17+
# Functional moonlight run - Disbaled till it can be fixed
18+
# tests/test_suites/llm/grpo-moonlight-16ba3b-4n4g-megatron.sh
1919

2020
# Functional VLM run
2121
tests/test_suites/vlm/vlm_grpo-qwen2.5-vl-3b-instruct-clevr-1n4g-dtensor2tp1.v1.sh

0 commit comments

Comments
 (0)