oyLabImaging/setup.py at master · ccb-hms/oyLabImaging · 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
from setuptools import setup, find_packages
from pathlib import Path

CU111_EXTRAS = [
    "cupy-cuda111 ; platform_system!='Darwin' and python_version<'3.10'",
    # PY38 - LINUX
    "torch@https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp38-cp38-linux_x86_64.whl ; platform_system=='Linux' and python_version=='3.8'",
    "torchvision@https://download.pytorch.org/whl/cu111/torchvision-0.9.1%2Bcu111-cp38-cp38-linux_x86_64.whl ; platform_system=='Linux' and python_version=='3.8'",
    "torchaudio@https://download.pytorch.org/whl/torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl ; platform_system=='Linux' and python_version=='3.8'",
    # PY38 - WINDOWS
    "torch@https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp38-cp38-win_amd64.whl ; platform_system=='Windows' and python_version=='3.8'",
    "torchvision@https://download.pytorch.org/whl/cu111/torchvision-0.9.1%2Bcu111-cp38-cp38-win_amd64.whl ; platform_system=='Windows' and python_version=='3.8'",
    "torchaudio@https://download.pytorch.org/whl/torchaudio-0.8.1-cp38-cp38-win_amd64.whl ; platform_system=='Windows' and python_version=='3.8'",
]


setup(
    name="oyLabImaging",
    version="0.2.10",
    description="data processing code for the Oyler-Yaniv lab @HMS",
    author="Alon Oyler-Yaniv",
    url="https://github.com/alonyan/oyLabImaging",
    packages=find_packages(include=["oyLabImaging", "oyLabImaging.*"]),
    python_requires=">=3.8, <3.9",
    long_description=Path("README.md").read_text(),
    long_description_content_type="text/markdown",
    dependency_links=["https://download.pytorch.org/whl/torch_stable.html"],
    install_requires=[
        "opencv-python-headless==4.7.0.68",
        "cellpose==0.7.2",
        "cloudpickle==1.6.0",
        "dill==0.3.4",
        "ipython>=7.27.0",
        "ipywidgets==7.6.5",
        "lap",
        "matplotlib>=3.3.4",
        "napari[pyqt5]==0.4.14",
        "pydantic<2", # pinned only because of napari
        "magicgui",
        "nd2>=0.8.0",
        "numba>=0.53.1",
        "numpy==1.23.5",
        "pandas>=1.2.4",
        "Pillow>=8.3.1",
        "poppy>=1.0.1",
        "scikit-image",
        "scikit-learn==1.0.2",
        "scipy>=1.6.2",
        "tqdm>=4.59.0",
        "zernike>=0.0.32",
        "multiprocess>=0.70",
        "jupyter>=1.0.0",
        "tensorflow-cpu==2.10.0 ; platform_machine!='arm64'",
        "stardist==0.8.3",
        "thin-plate-spline",
    ],
    extras_require={
        "cuda": CU111_EXTRAS,
        "test": ["pytest", "pytest-cov"],
    },
)