dlomix/setup.py at main · wilhelm-lab/dlomix · 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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import setuptools

with open("README.md", "r") as fh:
    long_description = fh.read()


def get_metadata():
    metadata = {}
    with open("src/dlomix/_metadata.py") as f:
        exec(f.read(), metadata)
    return metadata


# Load metadata
META_DATA = get_metadata()

tensorflow_extra_install = [
    "tensorflow>=2.13,<2.16",  # 2.16 introduces breaking changes and has Keras 3 as default
]

pytorch_extra_install = [
    "torch",
    "torchvision",
]

setuptools.setup(
    name=META_DATA["__package__"].lower(),
    version=META_DATA["__version__"],
    author=META_DATA["__author__"],
    author_email=META_DATA["__author_email__"],
    description=META_DATA["__description__"],
    long_description=long_description,
    long_description_content_type="text/markdown",
    url=META_DATA["__github_url__"],
    packages=setuptools.find_packages(where="src"),
    package_dir={"": "src"},
    include_package_data=True,
    package_data={"": ["data/processing/pickled_feature_dicts/*"]},
    python_requires=">=3.10",
    install_requires=[
        "datasets>=4.0.0",
        "huggingface_hub",
        "fpdf",
        "pandas",
        "numpy",
        "matplotlib",
        "scikit-learn",
        "pyarrow",
        "seaborn",
    ],
    extras_require={
        "dev": [
            "pytest >= 7.0.0",
            "pytest-cov",
            "black",
            "twine",
            "setuptools",
            "wheel",
            "pylint",
            *tensorflow_extra_install,
            *pytorch_extra_install,
        ],
        "wandb": [
            "wandb>=0.20.0",
        ],
        "tensorflow": tensorflow_extra_install,
        "tf": tensorflow_extra_install,
        "torch": pytorch_extra_install,
        "pytorch": pytorch_extra_install,
        "lightning": [
            "lightning",
        ],
    },
    classifiers=[
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
        "Topic :: Scientific/Engineering :: Bio-Informatics",
        "Development Status :: 1 - Planning",
        "Intended Audience :: Science/Research",
    ],
)