chore: PyTorch改为正式依赖,不再作为可选
This commit is contained in:
parent
531cff509d
commit
67b370364e
@ -19,6 +19,7 @@ dependencies = [
|
||||
"scikit-learn>=1.5.2",
|
||||
"xgboost>=2.1.0",
|
||||
"lightgbm>=4.5.0",
|
||||
"torch==2.5.1",
|
||||
|
||||
# 工具
|
||||
"openpyxl>=3.1.5",
|
||||
@ -27,15 +28,11 @@ dependencies = [
|
||||
]
|
||||
|
||||
[project.optional-dependencies]
|
||||
# PyTorch 为可选依赖(安装约 800MB,仅训练神经网络时需要)
|
||||
torch = [
|
||||
"torch==2.5.1",
|
||||
]
|
||||
dev = [
|
||||
# 测试工具
|
||||
"pytest>=7.0",
|
||||
"black>=22.0", # 代码格式化
|
||||
"mypy>=1.0", # 类型检查
|
||||
"black>=22.0",
|
||||
"mypy>=1.0",
|
||||
]
|
||||
|
||||
[build-system]
|
||||
|
||||
@ -7,6 +7,8 @@ pandas>=2.2.0
|
||||
xgboost>=2.1.0
|
||||
lightgbm>=4.5.0
|
||||
scikit-learn>=1.5.2
|
||||
torch==2.5.1
|
||||
|
||||
openpyxl>=3.1.5
|
||||
python-dotenv>=1.0.0
|
||||
requests>=2.31.0
|
||||
|
||||
Loading…
Reference in New Issue
Block a user