57 lines
1.1 KiB
TOML
57 lines
1.1 KiB
TOML
[project]
|
||
name = "cost-prediction"
|
||
version = "0.1.0"
|
||
description = "装备成本预测系统"
|
||
requires-python = ">=3.9,<3.12"
|
||
readme = "README.md"
|
||
license = {file = "LICENSE"}
|
||
|
||
dependencies = [
|
||
# Web框架
|
||
"flask>=3.1.0",
|
||
"flask-cors>=5.0.0",
|
||
|
||
# 数据处理
|
||
"numpy>=1.26.0,<2.0.0",
|
||
"pandas>=2.2.0",
|
||
|
||
# 机器学习
|
||
"scikit-learn>=1.5.2",
|
||
"xgboost>=2.1.0",
|
||
"lightgbm>=4.5.0",
|
||
|
||
# 工具
|
||
"openpyxl>=3.1.5",
|
||
"python-dotenv>=1.0.0",
|
||
"requests>=2.31.0",
|
||
]
|
||
|
||
[project.optional-dependencies]
|
||
# PyTorch 为可选依赖（安装约 800MB，仅训练神经网络时需要）
|
||
torch = [
|
||
"torch==2.5.1",
|
||
]
|
||
dev = [
|
||
# 测试工具
|
||
"pytest>=7.0",
|
||
"black>=22.0", # 代码格式化
|
||
"mypy>=1.0", # 类型检查
|
||
]
|
||
|
||
[build-system]
|
||
requires = ["setuptools>=61.0", "wheel"]
|
||
build-backend = "setuptools.build_meta"
|
||
|
||
[tool.pytest.ini_options]
|
||
testpaths = ["tests"]
|
||
python_files = ["test_*.py"]
|
||
|
||
[tool.black]
|
||
line-length = 88
|
||
target-version = ["py39", "py310", "py311"]
|
||
|
||
[tool.mypy]
|
||
python_version = "3.11"
|
||
warn_return_any = true
|
||
warn_unused_configs = true
|
||
|