# Required packages for the script
# Python version: 3.11.9

torch==2.3.0
numpy==1.26.4
scikit-learn==1.5.0
tqdm==4.66.4
pandas==2.2.2
xgboost==2.0.3
scipy==1.13.1

# For LLM few-shot
openai==1.35.7
google-generativeai==0.7.1
requests==2.32.3    

# Notes:
# 1. The torch version listed is for the base package.
#    If you require CUDA support, please install the appropriate PyTorch build
#    for your CUDA version. For example, for CUDA 12.1, you might install:
#    pip install torch==2.3.0+cu121 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
#    Refer to the official PyTorch website (pytorch.org) for the correct command.
#
# 2. These versions are known to be compatible with Python 3.11.
#    If you encounter issues, you might need to adjust versions based on your specific environment
#    or the exact versions used during the original development of the script.