Skip to content

Commit cf33363

Browse files
committed
Restore conftest before the AI additions to see if that fixes the stalling problems
Signed-off-by: John St. John <[email protected]>
1 parent dff8ed1 commit cf33363

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

bionemo-recipes/recipes/evo2_megatron/tests/bionemo/evo2/test_stop_and_go.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,10 @@
3636
@pytest.mark.parametrize(
3737
"tp_size,cp_size,dp_size,dp_rank_check,precision_recipe",
3838
[
39-
(1, 1, 1, False, "bf16_mixed"),
40-
(1, 1, 1, False, "bf16_with_fp8_current_scaling_mixed"),
39+
pytest.param(
40+
1, 1, 1, False, "bf16_mixed", marks=pytest.mark.skipif(bool(os.environ.get("CI")), reason="Skip in CI")
41+
),
42+
pytest.param(1, 1, 1, False, "bf16_with_fp8_current_scaling_mixed"),
4143
pytest.param(
4244
1,
4345
1,

0 commit comments

Comments
 (0)