From 2e35ad3f8f06f73c5bcaabc52920f220648f87da Mon Sep 17 00:00:00 2001 From: koko <1429659362@qq.com> Date: Wed, 10 Dec 2025 19:36:13 +0800 Subject: [PATCH] =?UTF-8?q?nep=E6=A1=86=E6=9E=B6=E9=87=8D=E6=9E=84=2004.pr?= =?UTF-8?q?edict?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/workflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/workflow.py b/src/workflow.py index b271e6f..d384b92 100644 --- a/src/workflow.py +++ b/src/workflow.py @@ -4,7 +4,7 @@ import shutil import logging import subprocess import re -from src.utils import load_yaml, run_cmd_with_log, parse_time_to_steps +from src.utils import load_yaml, run_cmd_with_log, parse_time_to_steps,Notifier from src.machine import MachineManager from src.state import StateTracker # 新增 from src.steps import MDStep, SelectStep, SCFStep, TrainStep @@ -21,7 +21,7 @@ class Workflow: self.template_dir = os.path.join(root_dir, "template") self.logger = logging.getLogger() - + self.notifier = Notifier() # 初始化状态追踪 os.makedirs(self.workspace, exist_ok=True) self.tracker = StateTracker(self.workspace)