from setuptools import setup, find_packages 

setup(
    name='abaffinity',
    version='0.1.2', 
    description='Predict Affinity against SARS-CoV-2 HR2 peptide',
    author='Faisal B. Ashraf',
    packages=find_packages(),
    include_package_data=True,  # Ensures all necessary files are included
    install_requires=[
        'torch',              # PyTorch
        'numpy',              # For numerical operations
        'matplotlib',         # For plotting graphs
        'textwrap3',          # For wrapping text
    ],
    python_requires=">=3.7",
)