File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
sagemaker-train/tests/integ/train Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 2222from sagemaker .train .common import TrainingType
2323
2424
25+ @pytest .mark .skip (reason = "Skipping GPU resource intensive test" )
2526def 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" )
6264def 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" )
100103def test_rlvr_trainer_nova_workflow (sagemaker_session ):
101104 """Test RLVR training workflow with Nova model."""
102105 import os
You can’t perform that action at this time.
0 commit comments