# This file is used to install the required packages for the PyTorch Inference Docker image.
# Using PyTorch CUDA 13.0 index (smaller than default PyPI, works on CPU and GPU instances)
--index-url https://download.pytorch.org/whl/cu130
--extra-index-url https://pypi.org/simple/
torch==2.9.1
pandas==2.3.3
awswrangler==3.14.0
lightgbm==4.6.0
mapie==1.2.0
joblib==1.5.3
chemprop==2.2.1   # Message Passing Neural Networks for molecular property prediction
rdkit==2025.9.3   # Cheminformatics toolkit for SMILES processing
fastapi==0.126.0  # FastAPI for serving the model
uvicorn==0.38.0   # ASGI server for running FastAPI
