setup.py

# -*- coding: utf-8 -*-
"""
    Setup file for BaumBauen.
    Use setup.cfg to configure your project.

    This file was generated with PyScaffold 3.2.3.
    PyScaffold helps you to put up the scaffold of your new Python project.
    Learn more under: https://pyscaffold.org/
"""
import sys

from pkg_resources import VersionConflict, require
from setuptools import setup

try:
    require('setuptools>=38.3')
except VersionConflict:
    print("Error: version of setuptools is too old (<38.3)!")
    sys.exit(1)

if __name__ == "__main__":
    setup(
        use_pyscaffold=True,
        install_requires=[
            'torch',
            'numpy',
            'scikit-learn',
            'scipy==1.4.1',
            'optuna',
            'pathlib',
            'tensorflow-cpu',
            'phasespace',
            'pytorch-ignite',
            'click',
            'notebook',
            'jupytext',
            'matplotlib',
            'h5py',
            'uproot',
            'pynvml',
            # The following are for Optuna visualisation
            'plotly',
            'kaleido',
        ]
    )