From 1b47093d1147bf1035adf3013a1ae0b4d216ba16 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Tue, 19 Sep 2023 16:02:06 +0200 Subject: [PATCH] test --- tests/exporters/onnx/test_onnx_export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/exporters/onnx/test_onnx_export.py b/tests/exporters/onnx/test_onnx_export.py index 4eefafa2d90..641e347c83a 100644 --- 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"