fix unit test

triton-inference-server · Jan 17, 2024 · 1b24043 · 1b24043
1 parent 44dc20f
commit 1b24043
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_quick_run_config_generator.py b/tests/test_quick_run_config_generator.py
@@ -155,7 +155,9 @@ def setUp(self):
         ):
             self._mock_models = [
                 ModelProfileSpec(
-                    ConfigModelProfileSpec(model_name="fake_model_name"),
+                    ConfigModelProfileSpec(
+                        model_name="fake_model_name", perf_analyzer_flags={}
+                    ),
                     MagicMock(),
                     MagicMock(),
                     MagicMock(),