collab-creatures/setup.py at main · BasisResearch/collab-creatures · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from setuptools import find_packages, setup

VERSION = "0.2.0"


# examples/tutorials/notebooks
DEV_REQUIRE = [
            "pytest==7.4.3",
            "pytest-cov",
            "pytest-xdist",
            "mypy",
            "black==24.8.0",
            "flake8",
            "isort==5.13.2",
            "nbval",
            "nbqa==1.9.0",
            "autoflake",
            "sphinx==7.1.2",
            "sphinxcontrib-bibtex",
            "sphinxcontrib-jquery",
            "sphinx_rtd_theme==1.3.0",
            "nbsphinx",
            "myst_parser",
            "pandoc",
        ]

setup(
    name="collab",
    version=VERSION,
    description="Tools for animal behavior foraging modeling.",
    packages=find_packages(include=["collab", "collab.*"]),
    author="Basis",
    url="https://www.basis.ai/",
    project_urls={
        "Documentation": "https://basisresearch.github.io/collab-creatures/getting_started.html",
        "Source": "https://github.com/BasisResearch/collaborative-intelligence",
    },
    install_requires=[
        "chirho @ git+https://github.com/BasisResearch/chirho.git#egg=chirho",
        "pyro-ppl>=1.8.6", "pandas==2.2.3", "plotly", "plotly.express",
        "torch>=2.2", "scipy", "scikit-learn",
        "matplotlib>=3.8.2", "dill", "torchdiffeq",
        "numpy",
        "jupyter", "graphviz", "matplotlib", "seaborn", "kaleido",
    ],
    extras_require={
        "dev": DEV_REQUIRE,
        },
    python_requires=">=3.10, <3.12",
    keywords="animal behavior, bayesian modeling, probabilistic programming, dynamical systems",
    license="Apache 2.0",
)