requirements-rl.txt

# SupplyMind RL dependencies (separate from HF Space requirements.txt)
# Install: pip install -r requirements-rl.txt

# Core ML
torch>=2.1.0
gymnasium==0.29.1

# RL frameworks
stable-baselines3==2.2.1
sb3-contrib==2.2.1
d3rlpy==2.3.0

# Transformers (for Decision Transformer backbone)
transformers>=4.36.0

# Dashboard
streamlit>=1.32.0
plotly>=5.18.0

# Explainability
shap>=0.43.0

# Statistical analysis
scipy>=1.11.0

# Real data
fredapi>=0.5.0

# Experiment tracking
mlflow>=2.10.0
wandb>=0.16.0

# LLM explainability (local)
ollama>=0.1.0

# RAG
chromadb==0.4.24
sentence-transformers>=2.2.0
PyPDF2>=3.0.0

# Forecasting
pytorch-forecasting==1.0.0
pytorch-lightning>=2.4.0

# Optimization
pymoo>=0.6.0
optuna>=3.4.0

# ONNX export
onnx>=1.15.0
onnxruntime>=1.16.0