修改模型更新脚本全局变量

ad/pai_flow_operator2.py

@@ -715,35 +715,35 @@ def validate_model_data_accuracy():
 
 if __name__ == '__main__':
     update_online_flow()
-    # start_time = int(time.time())
-    # functions = [update_online_flow, shuffle_table, shuffle_train_model, export_model, get_validate_model_data]
-    # function_names = [func.__name__ for func in functions]
-    #
-    # start_function = None
-    # if len(sys.argv) > 1:
-    #     start_function = sys.argv[1]
-    #     if start_function not in function_names:
-    #         print(f"指定的起始函数 {start_function} 不存在，请选择以下函数之一：{', '.join(function_names)}")
-    #         sys.exit(1)
-    #
-    # start_index = 0
-    # if start_function:
-    #     start_index = function_names.index(start_function)
-    #
-    # for func in functions[start_index:]:
-    #     if not func():
-    #         print(f"{func.__name__} 执行失败，后续函数不再执行。")
-    #         step_end_time = int(time.time())
-    #         elapsed = step_end_time - start_time
-    #         _monitor('error', f"DNN模型更新，{func.__name__} 执行失败，后续函数不再执行，请检查", start_time, elapsed, None)
-    #         break
-    # else:
-    #     print("所有函数都成功执行，可以继续下一步操作。")
-    #     result, msg, level, top10_msg = validate_model_data_accuracy()
-    #     if result:
-    #         # update_online_model()
-    #         print("success")
-    #     step_end_time = int(time.time())
-    #     elapsed = step_end_time - start_time
-    #     print(level, msg, start_time, elapsed, top10_msg)
-    #     _monitor(level, msg, start_time, elapsed, top10_msg)
+    start_time = int(time.time())
+    functions = [update_online_flow, shuffle_table, shuffle_train_model, export_model, get_validate_model_data]
+    function_names = [func.__name__ for func in functions]
+
+    start_function = None
+    if len(sys.argv) > 1:
+        start_function = sys.argv[1]
+        if start_function not in function_names:
+            print(f"指定的起始函数 {start_function} 不存在，请选择以下函数之一：{', '.join(function_names)}")
+            sys.exit(1)
+
+    start_index = 0
+    if start_function:
+        start_index = function_names.index(start_function)
+
+    for func in functions[start_index:]:
+        if not func():
+            print(f"{func.__name__} 执行失败，后续函数不再执行。")
+            step_end_time = int(time.time())
+            elapsed = step_end_time - start_time
+            _monitor('error', f"DNN模型更新，{func.__name__} 执行失败，后续函数不再执行，请检查", start_time, elapsed, None)
+            break
+    else:
+        print("所有函数都成功执行，可以继续下一步操作。")
+        result, msg, level, top10_msg = validate_model_data_accuracy()
+        if result:
+            # update_online_model()
+            print("success")
+        step_end_time = int(time.time())
+        elapsed = step_end_time - start_time
+        print(level, msg, start_time, elapsed, top10_msg)
+        _monitor(level, msg, start_time, elapsed, top10_msg)