from setuptools import setup, find_packages

setup(
    name="Disc_Hub_DIA",
    version="0.2.1",
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    install_requires=["numpy","matplotlib","pandas", "pyarrow==19.0.0",
                      "torch>=2.6.0","xgboost","scikit-learn"], #pip install torch==2.6.0+cu126 --extra-index-url https://download.pytorch.org/whl/cu126
    author="Yiwen Yu",
    description="A package for Disc Hub Data Integration Analysis",
)