Skip to content

Commit 506aeef

Browse files
committed
Skip RLVR integ tests
1 parent 8c07487 commit 506aeef

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

sagemaker-train/tests/integ/train/test_rlvr_trainer_integration.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
from sagemaker.train.common import TrainingType
2323

2424

25+
@pytest.mark.skip(reason="Skipping GPU resource intensive test")
2526
def test_rlvr_trainer_lora_complete_workflow(sagemaker_session):
2627
"""Test complete RLVR training workflow with LORA."""
2728

@@ -59,6 +60,7 @@ def test_rlvr_trainer_lora_complete_workflow(sagemaker_session):
5960
assert training_job.output_model_package_arn is not None
6061

6162

63+
@pytest.mark.skip(reason="Skipping GPU resource intensive test")
6264
def test_rlvr_trainer_with_custom_reward_function(sagemaker_session):
6365
"""Test RLVR trainer with custom reward function."""
6466

@@ -96,7 +98,8 @@ def test_rlvr_trainer_with_custom_reward_function(sagemaker_session):
9698
assert training_job.output_model_package_arn is not None
9799

98100

99-
@pytest.mark.skipif(os.environ.get('AWS_DEFAULT_REGION') != 'us-east-1', reason="Nova models only available in us-east-1")
101+
# @pytest.mark.skipif(os.environ.get('AWS_DEFAULT_REGION') != 'us-east-1', reason="Nova models only available in us-east-1")
102+
@pytest.mark.skip(reason="Skipping GPU resource intensive test")
100103
def test_rlvr_trainer_nova_workflow(sagemaker_session):
101104
"""Test RLVR training workflow with Nova model."""
102105
import os

0 commit comments

Comments
 (0)