Skip to content

Commit 6eac70c

Browse files
Joshua Townerjjtowner
authored andcommitted
test: use meta-textgeneration-llama-3-2-1b-instruct in integ tests
Switch to the same model used by ~80% of existing integ tests to avoid deprecation risk. Llama 3.2 1B is the most battle-tested model in the repo's test infrastructure.
1 parent 8a33376 commit 6eac70c

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

‎sagemaker-train/tests/integ/train/test_list_hyperparameters_integration.py‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ class TestListHyperparametersInteg:
2323

2424
def test_sft_lora_returns_expected_params(self):
2525
"""SFT LORA returns a FineTuningOptions with known hyperparameters."""
26-
hp = list_hyperparameters("huggingface-llm-qwen2-5-7b-instruct", "SFT", "LORA")
26+
hp = list_hyperparameters("meta-textgeneration-llama-3-2-1b-instruct", "SFT", "LORA")
2727

2828
assert isinstance(hp, FineTuningOptions)
2929
assert "learning_rate" in hp._specs
@@ -33,15 +33,15 @@ def test_sft_lora_returns_expected_params(self):
3333

3434
def test_dpo_lora_has_additional_params(self):
3535
"""DPO LORA returns params including adam_beta (not present in SFT)."""
36-
hp = list_hyperparameters("huggingface-llm-qwen2-5-7b-instruct", "DPO", "LORA")
36+
hp = list_hyperparameters("meta-textgeneration-llama-3-2-1b-instruct", "DPO", "LORA")
3737

3838
assert isinstance(hp, FineTuningOptions)
3939
assert "adam_beta" in hp._specs
4040
assert "learning_rate" in hp._specs
4141

4242
def test_rlvr_lora_returns_params(self):
4343
"""RLVR LORA returns FineTuningOptions with RL-specific params."""
44-
hp = list_hyperparameters("meta-textgeneration-llama-3-1-8b-instruct", "RLVR", "LORA")
44+
hp = list_hyperparameters("meta-textgeneration-llama-3-2-1b-instruct", "RLVR", "LORA")
4545

4646
assert isinstance(hp, FineTuningOptions)
4747
assert "learning_rate" in hp._specs
@@ -50,7 +50,7 @@ def test_rlvr_lora_returns_params(self):
5050
def test_accepts_enum_values(self):
5151
"""Accepts CustomizationTechnique and TrainingType enums."""
5252
hp = list_hyperparameters(
53-
"huggingface-llm-qwen2-5-7b-instruct",
53+
"meta-textgeneration-llama-3-2-1b-instruct",
5454
CustomizationTechnique.SFT,
5555
TrainingType.LORA,
5656
)
@@ -60,7 +60,7 @@ def test_accepts_enum_values(self):
6060

6161
def test_get_info_does_not_raise(self):
6262
"""get_info() runs without error on returned object."""
63-
hp = list_hyperparameters("huggingface-llm-qwen2-5-7b-instruct", "SFT", "LORA")
63+
hp = list_hyperparameters("meta-textgeneration-llama-3-2-1b-instruct", "SFT", "LORA")
6464
# Should print without raising
6565
hp.get_info("learning_rate")
6666

@@ -71,6 +71,6 @@ def test_invalid_model_raises(self):
7171

7272
def test_invalid_technique_raises(self):
7373
"""Technique not available for model raises an error."""
74-
# PPO is not available on Qwen 2.5
74+
# PPO is not available on Llama 3.2
7575
with pytest.raises(Exception):
76-
list_hyperparameters("huggingface-llm-qwen2-5-7b-instruct", "PPO", "LORA")
76+
list_hyperparameters("meta-textgeneration-llama-3-2-1b-instruct", "PPO", "LORA")

0 commit comments

Comments
 (0)