ECG-Bench/setup.py at main · willxxy/ECG-Bench · 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
85
from setuptools import find_packages, setup

INSTALL_REQUIRES = [
    "ruff",
    "pre-commit",
    "torch==2.6.0",
    "torchvision==0.21.0",
    "torchaudio==2.6.0",
    "transformers==4.56.0",
    "ecg_plot @ git+https://github.com/willxxy/ecg-plot",
    "accelerate==0.34.2",
    "einops==0.8.0",
    "h5py==3.12.1",
    "joblib==1.4.2",
    "matplotlib==3.9.2",
    "maturin==1.7.4",
    "numpy==1.26.4",
    "openai==2.3.0",
    "pandas==2.2.3",
    "peft==0.13.0",
    "PyWavelets==1.7.0",
    "PyYAML==6.0.2",
    "regex==2024.9.11",
    "requests==2.32.4",
    "scikit-learn==1.5.2",
    "scipy==1.14.1",
    "tqdm==4.66.5",
    "wandb==0.18.3",
    "wfdb==4.1.2",
    "seaborn==0.13.2",
    "nltk==3.9.1",
    "rouge==1.0.1",
    "sentencepiece==0.2.0",
    "sacremoses==0.1.1",
    "captum==0.7.0",
    "evaluate==0.4.3",
    "bert_score",
    "ninja==1.11.1.3",
    "beautifulsoup4==4.12.2",
    "imageio==2.27.0",
    "imgaug==0.4.0",
    "imutils==0.5.4",
    "opencv_python==4.6.0.66",
    "pillow==10.3.0",
    "scikit-image==0.21.0",
    "spacy==3.8.4",
    "validators==0.18.2",
    "html5lib==1.1",
    "qrcode==7.4.2",
    "gradio==5.31.0",
    "faiss-cpu==1.10.0",
    "termcolor==3.0.1",
    "pytest",
    "einops-exts==0.0.4",
    "open-clip-torch==3.2.0",
    "timm==1.0.22",
]

EXTRAS_REQUIRE = {
    "flash": [
        "flash-attn==2.7.4.post1",
    ],
    "judge": [
        "llm-blender @ git+https://github.com/yuchenlin/LLM-Blender.git",
        "trl[judges]",
    ],
}

# Add "all" as the union of everything
EXTRAS_REQUIRE["all"] = sorted(set(sum(EXTRAS_REQUIRE.values(), [])))

setup(
    name="ecg_bench",
    version="0.0.4",
    packages=find_packages(),
    url="https://github.com/willxxy/ECG-Bench",
    license="MIT",
    author="William Jongwon Han",
    author_email="wjhan@andrew.cmu.edu",
    description="Open source code of ECG-Bench",
    python_requires=">=3.10",
    install_requires=INSTALL_REQUIRES,
    extras_require=EXTRAS_REQUIRE,
    include_package_data=True,
)