element-deeplabcut/setup.py at main · AdaptiveMotorControlLab/element-deeplabcut · 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
54
55
56
57
58
59
60
61
62
from setuptools import setup, find_packages
from os import path

pkg_name = "element_deeplabcut"
here = path.abspath(path.dirname(__file__))

with open(path.join(here, "README.md"), "r") as f:
    long_description = f.read()

with open(path.join(here, pkg_name, "version.py")) as f:
    exec(f.read())

setup(
    name=pkg_name.replace("_", "-"),
    version=__version__,
    description="DataJoint Element for Continuous Behavior Tracking via DeepLabCut",
    long_description=long_description,
    long_description_content_type="text/markdown",
    author="DataJoint",
    author_email="info@datajoint.com",
    license="MIT",
    url=f'https://github.com/datajoint/{pkg_name.replace("_", "-")}',
    keywords="neuroscience behavior deeplabcut pose-estimation science datajoint",
    python_requires=">=3.10",  # Package tested with Python 3.10+ to match DLC 3.0.0rc13 at time of writing
    packages=find_packages(exclude=["contrib", "docs", "tests*"]),
    scripts=[],
    install_requires=[
        "datajoint==0.14.0",
        "graphviz",
        "pydot==1.4.2",
        "ipykernel==6.29.0",
        "ipywidgets==8.1.1",
    ],
    extras_require={
        "dlc_default": [
            "deeplabcut[superanimal]==3.0.0rc13",
            "dlclibrary>=0.1.0,<0.2.0",  # Pin to avoid ModelZoo download bug
        ],
        "dlc_apple_mchips": [
            "tensorflow-macos==2.12.0",
            "tensorflow-metal",
            "tables==3.7.0",
            "deeplabcut[superanimal]==3.0.0rc13",
            "dlclibrary>=0.1.0,<0.2.0",  # Pin to avoid ModelZoo download bug
        ],
        "dlc_gui": [
            "deeplabcut[gui]==3.0.0rc13",
            "dlclibrary>=0.1.0,<0.2.0",  # Pin to avoid ModelZoo download bug
        ],
        "elements": [
            "element-lab @ git+https://github.com/datajoint/element-lab.git",
            "element-animal @ git+https://github.com/datajoint/element-animal.git",
            "element-session @ git+https://github.com/datajoint/element-session.git",
            "element-interface @ git+https://github.com/datajoint/element-interface.git",
        ],
        "tests": [
            "pytest==7.4.4",
            "pytest-cov==4.1.0",
            "shutils",
        ],
    },
)