Enable the Recipe tests marked with @pytest.mark.skip(reason="Hyperpod recipe code unavailable" (#1642)

Author: nargokul

tests/unit/sagemaker/modules/train/sm_recipes/test_utils.py

@@ -48,8 +48,6 @@ def temporary_recipe():
         yield f.name
 
 
-# TODO: To be removed on 12/5/2024
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 def test_load_base_recipe_with_overrides(temporary_recipe, training_recipes_cfg):
     expected_epochs = 20
     expected_layers = 15
@@ -71,8 +69,6 @@ def test_load_base_recipe_with_overrides(temporary_recipe, training_recipes_cfg)
     )
 
 
-# TODO: To be removed on 12/5/2024
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 @pytest.mark.parametrize(
     "test_case",
     [
@@ -130,8 +126,6 @@ def test_load_base_recipe_types(
         assert mock_retrieve.call_args.args[0] == url
 
 
-# TODO: To be removed on 12/5/2024
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 @pytest.mark.parametrize(
     "test_case",
     [

tests/unit/sagemaker/modules/train/test_model_trainer.py

@@ -850,8 +850,6 @@ def mock_upload_data(path, bucket, key_prefix):
     )
 
 
-# TODO: To be removed on 12/5/2024
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 def test_model_trainer_gpu_recipe_full_init(modules_session):
     training_recipe = "training/llama/p4_hf_llama3_70b_seq8k_gpu"
     recipe_overrides = {"run": {"results_dir": "/opt/ml/model"}}

tests/unit/test_pytorch.py

@@ -844,7 +844,6 @@ def test_predictor_with_component_name(sagemaker_session, component_name):
     assert predictor._get_component_name() == component_name
 
 
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 def test_training_recipe_for_cpu(sagemaker_session):
     container_log_level = '"logging.INFO"'
 
@@ -878,7 +877,6 @@ def test_training_recipe_for_cpu(sagemaker_session):
         )
 
 
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 @pytest.mark.parametrize(
     "recipe, model",
     [
@@ -935,7 +933,6 @@ def test_training_recipe_for_gpu(sagemaker_session, recipe, model):
     assert pytorch.distribution.items() == expected_distribution.items()
 
 
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 def test_training_recipe_with_override(sagemaker_session):
     container_log_level = '"logging.INFO"'
 
@@ -973,7 +970,6 @@ def test_training_recipe_with_override(sagemaker_session):
     assert pytorch.image_uri == IMAGE_URI
 
 
-@pytest.mark.skip(reason="Hyperpod recipe code unavailable")
 def test_training_recipe_gpu_custom_source_dir(sagemaker_session):
     container_log_level = '"logging.INFO"'