test

huggingface · Sep 19, 2023 · 1b47093 · 1b47093
1 parent 2c063c0
commit 1b47093
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/exporters/onnx/test_onnx_export.py b/tests/exporters/onnx/test_onnx_export.py
@@ -570,7 +570,7 @@ def test_custom_export_official_model(self):
             assert "cross_attentions.0" in output_names
 
     @parameterized.expand(
-        grid_parameters({"fn_get_submodels": [None, fn_get_submodels_custom], "legacy": [True, False]})
+        grid_parameters({"fn_get_submodels": (None, fn_get_submodels_custom), "legacy": (True, False)})
     )
     def test_custom_export_trust_remote(self, test_name, fn_get_submodels, legacy):
         model_id = "fxmarty/tiny-mpt-random-remote-code"