Skip to content

Commit 16de22f

Browse files
authored
Fix KeyError in failing nightly (#1866)
SUMMARY: Fixes: https://github.com/neuralmagic/llm-compressor-testing/actions/runs/17901928706/job/50896401512 Restores previous behavior before #1844 merged. Signed-off-by: Fynn Schmitt-Ulms <[email protected]>
1 parent 0b5272e commit 16de22f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def test_oneshot_and_finetune_gpu(config, tmp_path):
101101
model = config["model"]
102102
dataset = config["dataset"]
103103
recipe = config["recipe"]
104-
dataset_config_name = config["dataset_config_name"]
104+
dataset_config_name = config.get("dataset_config_name")
105105
num_train_epochs = config["num_train_epochs"]
106106
concat_txt = config["concat_txt"]
107107
output = tmp_path / "finetune_output"

0 commit comments

Comments
 (0)