numpy<2
matplotlib
tqdm
pyyaml
transformers>=4.36.0
huggingface-hub>=0.20.0
accelerate>=0.25.0
pydantic>=2.0.0
torch>=2.0.0
streamlit>=1.28.0

[all]
collm-hep[dev,langchain]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
ruff>=0.1.0
mypy>=1.0.0

[langchain]
langchain>=0.1.0
langchain-huggingface>=0.0.1
