diff --git a/tests/integration/test_lists/test-db/l0_a10.yml b/tests/integration/test_lists/test-db/l0_a10.yml index 6bcc615360ff..e1452d6eecd5 100644 --- a/tests/integration/test_lists/test-db/l0_a10.yml +++ b/tests/integration/test_lists/test-db/l0_a10.yml @@ -245,6 +245,7 @@ l0_a10: - llmapi/test_llm_examples.py::test_llmapi_quickstart_atexit - unittest/test_model_runner_cpp.py - unittest/llmapi/test_build_cache.py + - unittest/others/test_builder.py - unittest/trt/attention/test_gpt_attention_IFB.py - unittest/trt/attention/test_gpt_attention_no_cache.py - condition: diff --git a/tests/unittest/others/test_builder.py b/tests/unittest/others/test_builder.py index 1812c7338810..29f9675d6981 100644 --- a/tests/unittest/others/test_builder.py +++ b/tests/unittest/others/test_builder.py @@ -57,7 +57,9 @@ def test_basic_builder_flow(self): assert engine is not None refit_engine = builder.refit_engine(network, engine) assert refit_engine is not None - builder.save_config(builder_config, tempfile.mktemp()) + with tempfile.TemporaryDirectory() as tmpdir: + config_path = f"{tmpdir}/builder_config.json" + builder.save_config(builder_config, config_path) def test_top_level_dont_have_functional_apis(self): # This did not check all the functional apis, but should already prevent