Skip to content

Commit 0d31d46

Browse files
committed
Update recipe loader schema expectations
Signed-off-by: Shengliang Xu <shengliangx@nvidia.com>
1 parent 5969cb3 commit 0d31d46

1 file changed

Lines changed: 0 additions & 2 deletions

File tree

tests/unit/recipe/test_loader.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -563,7 +563,6 @@ def test_import_entry_element_schema_appends(tmp_path):
563563
recipe = load_recipe(recipe_file)
564564
assert _entry_to_dict(recipe.quantize["quant_cfg"][0]) == {
565565
"quantizer_name": "*",
566-
"cfg": None,
567566
"enable": False,
568567
}
569568

@@ -977,7 +976,6 @@ def test_import_mixed_tree(tmp_path):
977976
assert _entry_to_dict(data["quant_cfg"][1]) == {
978977
"quantizer_name": "*lm_head*",
979978
"enable": False,
980-
"cfg": None,
981979
}
982980

983981

0 commit comments

Comments
 (0)