open_pub/Uni-Lab-OS
1
0
Fork 0
mirror of https://github.com/dptech-corp/Uni-Lab-OS.git synced 2026-02-04 21:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Build 2

Browse Source
This commit is contained in:
Xuwznln
2026-01-27 13:45:32 +08:00
parent ab05b858e1
commit 6db7fbd721
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
unilabos/app/main.py
View File

@@ -217,6 +217,8 @@ def main():
# 环境检查 - 检查并自动安装必需的包 (可选) # 环境检查 - 检查并自动安装必需的包 (可选)
skip_env_check = args_dict.get("skip_env_check", False) skip_env_check = args_dict.get("skip_env_check", False)
check_mode = args_dict.get("check_mode", False)
if not skip_env_check: if not skip_env_check:
from unilabos.utils.environment_check import check_environment from unilabos.utils.environment_check import check_environment
@@ -229,6 +231,8 @@ def main():
# 加载配置文件，优先加载config，然后从env读取 # 加载配置文件，优先加载config，然后从env读取
config_path = args_dict.get("config") config_path = args_dict.get("config")
if check_mode:
args_dict["working_dir"] = os.path.abspath(os.getcwd())
# 当 skip_env_check 时，默认使用当前目录作为 working_dir # 当 skip_env_check 时，默认使用当前目录作为 working_dir
if skip_env_check and not args_dict.get("working_dir") and not config_path: if skip_env_check and not args_dict.get("working_dir") and not config_path:
working_dir = os.path.abspath(os.getcwd()) working_dir = os.path.abspath(os.getcwd())
@@ -273,9 +277,11 @@ def main():
print_status(f"已创建 local_config.py 路径： {config_path}", "info") print_status(f"已创建 local_config.py 路径： {config_path}", "info")
else: else:
os._exit(1) os._exit(1)
# 加载配置文件
# 加载配置文件 (check_mode 跳过)
print_status(f"当前工作目录为 {working_dir}", "info") print_status(f"当前工作目录为 {working_dir}", "info")
load_config_from_file(config_path) if not check_mode:
load_config_from_file(config_path)
# 根据配置重新设置日志级别 # 根据配置重新设置日志级别
from unilabos.utils.log import configure_logger, logger from unilabos.utils.log import configure_logger, logger
@@ -331,12 +337,7 @@ def main():
machine_name = "".join([c if c.isalnum() or c == "_" else "_" for c in machine_name]) machine_name = "".join([c if c.isalnum() or c == "_" else "_" for c in machine_name])
BasicConfig.machine_name = machine_name BasicConfig.machine_name = machine_name
BasicConfig.vis_2d_enable = args_dict["2d_vis"] BasicConfig.vis_2d_enable = args_dict["2d_vis"]
# Check mode 处理
check_mode = args_dict.get("check_mode", False)
BasicConfig.check_mode = check_mode BasicConfig.check_mode = check_mode
if check_mode:
print_status("Check mode 启用，将进行 complete_registry 检查", "info")
from unilabos.resources.graphio import ( from unilabos.resources.graphio import (
read_node_link_json, read_node_link_json,