@@ -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
119119def 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
157157def test_save_config_minimal (
158158 config_fp32 : dict ,
0 commit comments