setup.py

import pathlib

import setuptools

_here = pathlib.Path(__file__).resolve().parent
print(_here)

name = "src"
author = "Aidan Scannell,  Riccardo Mereu"
author_email = "scannell.aidan@gmail.com, riccardomereu5@gmail.com"
description = (
    "Sparse function-space representation (SFR) of neural networks in PyTorch."
)

with open(_here / "README.md", "r") as f:
    readme = f.read()

url = ""

license = "Apache-2.0"

classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "Intended Audience :: Developers",
    "Intended Audience :: Information Technology",
    "License :: OSI Approved :: Apache Software License",
    "Natural Language :: English",
    "Programming Language :: Python :: 3",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: Information Analysis",
    "Topic :: Scientific/Engineering :: Mathematics",
]
keywords = [
    "deep-learning",
    "machine-learning",
    "probabilistic-modelling",
    "bayesian-inference",
    "approximate-inference",
    "laplace-approximation",
    "bayesian-deep-learning",
    "bayesian-neural-networks",
    "gaussian-processes",
    "continual-learning",
    "model-based-reinforcement-learning",
    "planning",
]

python_requires = ">=3.8"

install_requires = [
    "torch==2.0.0",
    "torchvision==0.15.1",
    "torchtyping==0.1.4",
    "matplotlib==3.5.1",
    "numpy==1.24.2",
    "ax-platform==0.3.3",
    # "laplace-torch", # install separately due to backpacpk-for-pytorch requiring torch 1.
]
extras_require = {
    "dev": [
        "black==23.3.0",
        "pre-commit==3.2.2",
        "pyright==1.1.301",
        "isort==5.12.0",
        "pyflakes==3.0.1",
        "pytest==7.2.2",
    ],
    "experiments": [
        "wandb==0.15",
        "hydra-core==1.3.2",
        "hydra-submitit-launcher==1.2.0",
        "hydra-colorlog==1.2",
        "jupyter==1.0.0",
        "netcal==1.3.5",  # needed for computing ECE
        "mujoco==2.3.3",
        "dm_control==1.0.11",  # deepmind control suite
        "opencv-python==4.7.0.72",
        "moviepy==1.0.3",  # rendering
        "tikzplotlib",
        # "tikzplotlib==0.10.1",
        # "gpytorch==1.9.1",  # for RL SVGP experiments
        "gym[classic_control]==0.26.2",
        "pandas",  # for making UCI table
        "seaborn",
        # "plotly==5.1.0",
        "uci_datasets @ git+https://github.com/treforevans/uci_datasets",
        ##### For MuJoCo rendering #####
        # "glew",
        # "mesalib",
        # "mesa-libgl-cos6-x86_64",
        # "glfw3",
    ],
    "figures": [  # needed to get the HMC result for figure 1
        "hamiltorch @ git+https://github.com/AdamCobb/hamiltorch",
    ],
}

setuptools.setup(
    name=name,
    version="0.1.0",
    author=author,
    author_email=author_email,
    maintainer=author,
    maintainer_email=author_email,
    description=description,
    keywords=keywords,
    long_description=readme,
    long_description_content_type="text/markdown",
    url=url,
    license=license,
    classifiers=classifiers,
    zip_safe=False,
    python_requires=python_requires,
    install_requires=install_requires,
    extras_require=extras_require,
    # packages=setuptools.find_namespace_packages(),
    packages=setuptools.find_packages(exclude=["notebooks", "paper"]),
)