From fac518b96659f40cac2e17c22557130bc0dc75de Mon Sep 17 00:00:00 2001 From: Xiaoyu Date: Fri, 22 Mar 2024 07:07:24 +0000 Subject: [PATCH] fix test --- onnxruntime/test/python/transformers/test_optimizer_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/test/python/transformers/test_optimizer_utils.py b/onnxruntime/test/python/transformers/test_optimizer_utils.py index 70d26cf7b4af3..17363829eab1a 100644 --- a/onnxruntime/test/python/transformers/test_optimizer_utils.py +++ b/onnxruntime/test/python/transformers/test_optimizer_utils.py @@ -13,7 +13,7 @@ class TestOptimizerUtils(unittest.TestCase): def test_extract_raw_data_from_model(self): - model = self._get_model_proto_with_raw_data() + model = self._get_model_proto_with_raw_data(False) external_names, external_values = extract_raw_data_from_model(model) self.assertEqual(list(external_names), ["inputs"]) self.assertEqual(len(external_values), 1)