diff --git a/pyproject.toml b/pyproject.toml index 12ee67e..2c9ceae 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -23,16 +23,14 @@ dependencies = [ # 机器学习 "scikit-learn>=1.5.2", + "xgboost>=2.1.0", + "lightgbm>=4.5.0", "torch==2.5.1", - "torchvision==0.20.1", - "torchaudio==2.5.1", - "xgboost>=2.1.0", # 添加 XGBoost - "lightgbm>=4.5.0", # 添加 LightGBM # 工具 - "openpyxl>=3.1.5", # Excel支持 - "python-dotenv>=1.0.0", # 环境变量 - "requests>=2.31.0", # API测试 + "openpyxl>=3.1.5", + "python-dotenv>=1.0.0", + "requests>=2.31.0", ] [project.optional-dependencies]