setup.py

import runpy
from setuptools import setup, find_packages

__version__ = runpy.run_path("torch_em/__version__.py")["__version__"]


# NOTE requirements are not all available via pip, you need to use conda,
# see 'environment_gpu.yaml' / 'environment_cpu.yaml'
requires = [
    "torch",
    "h5py"
]


setup(
    name="torch_em",
    packages=find_packages(exclude=["test"]),
    version=__version__,
    author="Constantin Pape",
    install_requires=requires,
    url="https://github.com/constantinpape/torch-em",
    license="MIT",
    entry_points={
        "console_scripts": [
            "torch_em.train_2d_unet = torch_em.cli:train_2d_unet",
            "torch_em.train_3d_unet = torch_em.cli:train_3d_unet",
            "torch_em.predict = torch_em.cli:predict",
            "torch_em.predict_with_tiling = torch_em.cli:predict_with_tiling",
            "torch_em.export_bioimageio_model = torch_em.util.modelzoo:main",
            "torch_em.validate_checkpoint = torch_em.util.validation:main",
            "torch_em.submit_slurm = torch_em.util.submit_slurm:main",
        ]
    }
)