#
# Full "complete" environment to mirror scripts/setup_uv.sh
# Install with: pip install -r requirements.txt
#

# Core runtime deps
numpy>=1.22,<2.0
termcolor>=1.1.0
scikit-learn
pandas>=1.4.3
tabulate>=0.8.10
torch>=1.12.0
pytorch-lightning>=1.6.3
tensorboard>=2.9.1

# Dev tools
pytest>=6.0
pytest-cov
black
isort
flake8
mypy
pre-commit

# Docs stack
mkdocs>=1.4.0
mkdocs-material
mkdocs-jupyter
matplotlib
seaborn

# Notebook stack
jupyter>=1.0.0
ipykernel
matplotlib>=3.5.0
seaborn>=0.11.0
plotly>=5.0.0

# Benchmarking
tqdm>=4.64.0
torch-scatter>=2.0.9
torchmetrics>=0.9.2

# Additional "complete" extras
absl-py>=1.2.0
aiohttp>=3.8.1
PyYAML>=6.0
requests>=2.28.1
scipy>=1.8.1

# Install project in editable mode
-e .
