torch>=2.0
numpy>=1.24
tqdm
faiss-cpu>=1.9.0
scikit-learn>=1.5.2
huggingface_hub

[dev]
torch==2.3.1
numpy==1.26.4
scikit-learn==1.5.2
econml==0.15.1
huggingface_hub==0.32.4
tqdm==4.67.1
faiss-cpu==1.9.0
pytest==8.4.0
black==25.1.0
isort==6.0.1
jupyter==1.1.1
ipython==8.37.0
pandas==2.3.0
pytest-cov==6.1.1
matplotlib==3.10.3
scikit-uplift==0.5.1
python-dotenv==1.1.0
FLAML[automl]==2.3.5
