# GAT-LSTM TMP预测模型依赖包
# Python 3.8+

# 深度学习框架
torch>=1.10.0
torch-geometric>=2.0.0

# 数据处理
pandas>=1.3.0
numpy>=1.21.0
scikit-learn>=1.0.0

# Web框架
fastapi>=0.95.0
uvicorn[standard]>=0.20.0
pydantic>=1.10.0

# 信号处理
scipy>=1.7.0
PyWavelets>=1.1.1

# 工具库
joblib>=1.1.0
tqdm>=4.62.0
colorlog>=6.6.0
pyyaml>=6.0

# 可选：CUDA支持（根据需要安装）
# torch-cu117  # CUDA 11.7
# torch-cu118  # CUDA 11.8