diff --git a/tests/test_quick_run_config_generator.py b/tests/test_quick_run_config_generator.py index 544a7bd5a..3623a978b 100755 --- 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(),