Update model_analyzer/config/generate/perf_analyzer_config_generator.py

Co-authored-by: Hyunjae Woo <[email protected]>
triton-inference-server · Oct 11, 2023 · b1eed54 · b1eed54
1 parent c69b577
commit b1eed54
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/model_analyzer/config/generate/perf_analyzer_config_generator.py b/model_analyzer/config/generate/perf_analyzer_config_generator.py
@@ -360,9 +360,8 @@ def _update_perf_config_for_llm_model(
         perf_config.update_config({"input-data": self._input_json_filename})
 
     def _write_modified_input_dict_to_file(self, modified_input_dict: Dict) -> None:
-        temp_input_data = open(self._input_json_filename, "w")
-        json.dump(modified_input_dict, temp_input_data)
-        temp_input_data.close()
+        with open(self._input_json_filename, "w") as f:
+            json.dump(modified_input_dict, f)
 
     def _create_non_parameter_perf_config_values(self) -> dict:
         perf_config_values = {