File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -53,6 +53,7 @@ def test_model_quantized(
5353 with pytest .raises (RuntimeError ):
5454 qmodel_prep (model_quantized , sample_input_fp32 , config_fp32 )
5555
56+
5657def test_recipe_not_present (
5758 wrong_recipe_name : str ,
5859):
Original file line number Diff line number Diff line change @@ -84,6 +84,7 @@ def test_save_config_wanted_pairs(
8484
8585 delete_config ()
8686
87+
8788def test_save_config_with_qcfg_save (
8889 config_fp32 : dict ,
8990 save_list : list ,
@@ -116,6 +117,7 @@ def test_save_config_with_qcfg_save(
116117 delete_config ()
117118 del config_fp32 ["keys_to_save" ]
118119
120+
119121def test_save_config_with_recipe_save (
120122 config_fp32 : dict ,
121123 save_list : list ,
@@ -154,6 +156,7 @@ def test_save_config_with_recipe_save(
154156 delete_config ()
155157 delete_config ("keys_to_save.json" )
156158
159+
157160def test_save_config_minimal (
158161 config_fp32 : dict ,
159162):
@@ -198,6 +201,7 @@ def test_double_qconfig_save(
198201
199202 delete_config ()
200203
204+
201205def test_qconfig_save_list_as_dict (
202206 config_fp32 : dict ,
203207):
@@ -247,7 +251,6 @@ def test_qconfig_save_recipe_as_dict(
247251 }
248252 save_json (save_dict , file_path = "keys_to_save.json" )
249253
250-
251254 with pytest .raises (ValueError ):
252255 qconfig_save (config_fp32 , recipe = "keys_to_save.json" , minimal = True )
253256
@@ -265,7 +268,7 @@ def test_qconfig_load_with_recipe_as_list(
265268 """
266269 delete_config ()
267270
268- config_list = list ( config_fp32 .keys () )
271+ config_list = list (config_fp32 .keys ())
269272
270273 save_json (config_list , file_path = "qcfg.json" )
271274
You can’t perform that action at this time.
0 commit comments