setup.py

from pathlib import Path
from setuptools import find_packages, setup

# Current directory
HERE = Path(__file__).resolve().parent

README = (HERE / "README.md").read_text()
DESC = ('PyTorch-based differentiable procedural material library, supporting material graph '
        'translation and node parameter optimization against user-input appearance captures')

REQUIRES_PYTHON = '>=3.7.0'
VERSION = '0.2.1'

# Run setup
setup(
    name='diffmat',
    version=VERSION,
    description=DESC,
    long_description=README,
    long_description_content_type="text/markdown",
    author='Beichen Li; Liang Shi',
    author_email='beichen@mit.edu; liangs@mit.edu',
    python_requires=REQUIRES_PYTHON,
    url="TODO",
    keywords='procedural material, differentiable computation, appearance modeling, SVBRDF map, '
             'gradient descent, optimization',
    packages=find_packages(),
    install_requires=[
        'torch>=1.11.0',
        'torchvision>=0.12.0',
        'numpy>=1.21.5',
        'scipy>=1.7.3',
        'imageio>=2.9.0',
        'pandas>=1.3.5',
        'pyyaml>=6.0',
        'taichi>=1.3.0'
    ],
    include_package_data=True,
    license='Custom License',
    classifiers=[
        'Development Status :: 2 - Pre-Alpha',
        'Topic :: Multimedia :: Graphics',
        'Intended Audience :: Product/Research',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: 3.9',
        'Programming Language :: Python :: 3.10',
    ],
)