dantis/setup.py at master · kdis-lab/dantis · 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
86
87
88
89
90
from setuptools import setup, find_packages
from codecs import open
from os import path

here = path.abspath(path.dirname(__file__))

with open(path.join(here, "README.md"), encoding="utf-8") as f:
    long_description = f.read()

# =========================
# Base requirements (library)
# =========================
install_requires = [
    "numpy>=1.20",
    "pyod>=1.1",
    "xgboost>=2.0",
    "combo>=0.1",
    "suod>=0.1.1",
    "tensorflow>=2.13",
    "torch>=2.1.2",
    "stumpy>=1.12.0",
    "statsmodels>=0.13.5",
    "saxpy==1.0.1.dev167",
    "pomegranate==1.1.2",
    "fuzzy-c-means>=1.7.2",
    "river>=0.22.0",
    "PyWavelets>=1.8.0",
]

# =========================
# Optional requirements (extras)
# =========================
extras_require = {
    # Install GUI/app dependencies:
    #   pip install "dantis[app]"
    "app": [
        "PyQt5==5.15.11",
        "arff==0.9",
    ],
}

setup(
    # Project name
    # $ pip install dantis
    name="dantis",

    # Version
    version="0.0.1",

    # Description
    description=(
        "DANTIS brings together **90+ anomaly detection algorithms** (from classical "
        "statistical methods and machine learning to deep learning) under a **unified "
        "scikit-learn–style API**. It consolidates models from diverse libraries and "
        "includes proprietary algorithms developed by our team."
    ),

    # Long description (README)
    long_description=long_description,
    long_description_content_type="text/markdown",

    # URL
    url="https://github.com/kdis-lab/dantis",

    # Author
    author="DaSCI, KDIS Lab, SCI2S",

    # Author email
    author_email="",

    # Keywords
    keywords=[
        "Anomaly Detection",
        "Time Series",
        "Machine Learning",
        "Deep Learning",
        "Scikit-learn",
        "Data Science",
    ],

    # Packages
    package_dir={"": "./"},
    packages=find_packages(where="./", exclude=["app", "docs", "tests", "examples", "src", "setup.py"]),
    include_package_data=True,
    package_data={},

    # Requirements
    install_requires=install_requires,
    extras_require=extras_require,
)