diff --git a/tests/service/test_autotune_service.py b/tests/service/test_autotune_service.py index 2f5cddd03..f0b034b43 100644 --- a/tests/service/test_autotune_service.py +++ b/tests/service/test_autotune_service.py @@ -292,7 +292,7 @@ def test_autotune_service(self): mock_objs = [] pool = multiprocessing.Pool(nprocs * len(model_dict)) - results = dict([(key, []) for key in model_dict.keys()]) + results = {key: [] for key in model_dict.keys()} for (model_name, (tensor_list, spans)) in model_dict.items(): pg_init_method = "file:///tmp/.bagua.unittest.autotune.{}".format(model_name) for i in range(nprocs):