Skip to content

Commit fdda6d9

Browse files
committed
test: Changed save_list to keys_to_save
Signed-off-by: Brandon Groth <[email protected]>
1 parent a5413dc commit fdda6d9

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

tests/models/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -798,10 +798,10 @@ def config_fp16(request):
798798
qconfig["nbits_w"] = 16
799799
return qconfig
800800

801-
save_list_params = [
801+
keys_to_save_params = [
802802
["qa_mode", "qw_mode", "nbits_a", "nbits_w", "qskip_layer_name"],
803803
]
804-
@pytest.fixture(scope="session", params=save_list_params)
804+
@pytest.fixture(scope="session", params=keys_to_save_params)
805805
def save_list(request):
806806
"""
807807
Generate a save list for testing user-requested save config.

tests/models/test_saveconfig.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def test_save_config_with_qcfg_save(
9696
save_list (list): List of variables to save in a quantized config.
9797
"""
9898
delete_config()
99-
config_fp32["save"] = save_list
99+
config_fp32["keys_to_save"] = save_list
100100

101101
qconfig_save(config_fp32, minimal=False)
102102

@@ -114,7 +114,7 @@ def test_save_config_with_qcfg_save(
114114
assert loaded_config.get(key) == config_fp32.get(key)
115115

116116
delete_config()
117-
del config_fp32["save"]
117+
del config_fp32["keys_to_save"]
118118

119119
def test_save_config_with_recipe_save(
120120
config_fp32: dict,
@@ -129,13 +129,13 @@ def test_save_config_with_recipe_save(
129129
"""
130130
# Delete both qcfg and the save.json before starting
131131
delete_config()
132-
delete_config("save.json")
132+
delete_config("keys_to_save.json")
133133

134134
# Save new "save.json"
135-
save_path = "save_list.json"
135+
save_path = "keys_to_save.json"
136136
save_json(save_list, file_path=save_path)
137137

138-
qconfig_save(config_fp32, recipe="save_list")
138+
qconfig_save(config_fp32, recipe="keys_to_save")
139139

140140
# Check that saved qcfg matches
141141
loaded_config = load_json()
@@ -152,7 +152,7 @@ def test_save_config_with_recipe_save(
152152
assert loaded_config.get(key) == config_fp32.get(key)
153153

154154
delete_config()
155-
delete_config("save.json")
155+
delete_config("keys_to_save.json")
156156

157157
def test_save_config_minimal(
158158
config_fp32: dict,

0 commit comments

Comments
 (0)