From c4a596df9e7b1a2bc2917f4e9db78b44aa68a2d5 Mon Sep 17 00:00:00 2001
From: Isotr0py <2037008807@qq.com>
Date: Fri, 10 Mar 2023 13:44:16 +0800
Subject: [PATCH] replace unsafe eval() with ast

---
 library/train_util.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/library/train_util.py b/library/train_util.py
index d4ab6d58..dc0269cc 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1,6 +1,7 @@
 # common functions for training
 
 import argparse
+import ast
 import importlib
 import json
 import re
@@ -1886,11 +1887,11 @@ def get_scheduler_fix(args,optimizer: Optimizer):
         if value[i].lower() == "true" or value[i].lower() == "false":
           value[i] = (value[i].lower() == "true")
         else:
-          value[i] = eval(value[i]) # warning: not safe!!!
+          value[i] = ast.literal_eval(value[i])
       if len(value) == 1:
         value = value[0]
       else:
-        value = tuple(value)  # some may use list?
+        value = list(value)  # some may use list?
 
       lr_scheduler_kwargs[key] = value