fix test

microsoft · Mar 22, 2024 · fac518b · fac518b
1 parent e00fd13
commit fac518b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)