Skip to content

Commit 80f8cd3

Browse files
committed
Updated test id names.
Signed-off-by: meetkuma <[email protected]>
1 parent b0aeb4d commit 80f8cd3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/finetune/test_finetune.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def download_alpaca():
5454
1.00232565, # expected_train_metric
5555
0.02060609, # expected_eval_loss
5656
1.02081990, # expected_eval_metric
57-
id="llama_config", # config name
57+
id="llama_config_samsum", # config name
5858
),
5959
pytest.param(
6060
"meta-llama/Llama-3.2-1B", # model_name
@@ -72,7 +72,7 @@ def download_alpaca():
7272
1.0043447, # expected_train_metric
7373
0.0117334, # expected_eval_loss
7474
1.0118025, # expected_eval_metric
75-
id="llama_config", # config name
75+
id="llama_config_gsm8k", # config name
7676
),
7777
pytest.param(
7878
"meta-llama/Llama-3.2-1B", # model_name
@@ -90,7 +90,7 @@ def download_alpaca():
9090
1.0006101, # expected_train_metric
9191
0.0065296, # expected_eval_loss
9292
1.0065510, # expected_eval_metric
93-
id="llama_config", # config name
93+
id="llama_config_alpaca", # config name
9494
),
9595
pytest.param(
9696
"google-bert/bert-base-uncased", # model_name
@@ -108,7 +108,7 @@ def download_alpaca():
108108
0.55554199, # expected_train_metric
109109
0.00738618, # expected_eval_loss
110110
0.70825195, # expected_eval_metric
111-
id="bert_config", # config name
111+
id="bert_config_imdb", # config name
112112
),
113113
]
114114

0 commit comments

Comments
 (0)