setup.py

import sys
from io import open
from os import path

from setuptools import find_packages, setup

here = path.abspath(path.dirname(__file__))
sys.path.insert(0, path.join(here, "tuneupcommon"))

# Get the long description from the README file
with open(path.join(here, "README.md"), encoding="utf-8") as f:
    long_description = f.read()

setup(
    name="tuneupcommon",
    version="0.1.0",
    description="TuneUp",
    url="https://github.com/weihua916/tuneup",
    author="Weihua Hu",
    author_email="weihua916@gmail.com",
    keywords=[
        "pytorch",
        "graph machine learning",
        "graph representation learning",
        "graph neural networks",
    ],
    long_description=long_description,
    long_description_content_type="text/markdown",
    install_requires=[
        "torch>=1.2.0",
        "numpy>=1.16.0",
        "torch_geometric>=2.0.0",
    ],
    license="MIT",
    packages=find_packages(
        exclude=["dataset_preprocessing", "assests", "nodecls", "recsys", "linkpred"]
    ),
    package_data={"tuneupcommon": []},
    include_package_data=True,
    classifiers=[
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "Intended Audience :: Science/Research",
        "Programming Language :: Python :: 3.8",
        "License :: OSI Approved :: MIT License",
    ],
)