Skip to content

Commit db5fd47

Browse files
fix: replace eval with ast.literal_eval
1 parent 50a1ded commit db5fd47

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

iotdb-core/ainode/iotdb/ainode/core/config.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
# specific language governing permissions and limitations
1616
# under the License.
1717
#
18+
import ast
1819
import os
1920
import re
2021

@@ -337,7 +338,9 @@ def _load_config_from_file(self) -> None:
337338

338339
if "ain_inference_model_mem_usage_map" in config_keys:
339340
self._config.set_ain_inference_model_mem_usage_map(
340-
eval(file_configs["ain_inference_model_mem_usage_map"])
341+
ast.literal_eval(
342+
file_configs["ain_inference_model_mem_usage_map"]
343+
)
341344
)
342345

343346
if "ain_inference_memory_usage_ratio" in config_keys:

0 commit comments

Comments
 (0)