waveflow/setup.py at main · nqmn/waveflow · 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
#!/usr/bin/env python3
"""Legacy setuptools entry point.

The canonical project metadata lives in ``pyproject.toml``. This file keeps
older setuptools workflows working and mirrors the package discovery/console
entry point needed for editable installs.
"""

from setuptools import setup, find_packages

BASE_REQUIRES = [
    "numpy",
    "scipy",
    "pyyaml",
]

EXTRAS_REQUIRE = {
    "web": [
        "flask",
        "waitress",
    ],
    "vision": [
        "opencv-python",
    ],
    "optimization": [
        "cvxpy",
        "scs",
    ],
    "plot": [
        "matplotlib",
    ],
    "terminal": [
        "typer",
        "rich",
    ],
    "ml": [
        "torch>=1.9.0",
        "scikit-learn",
    ],
    "dev": [
        "pytest>=6.0",
        "pytest-cov",
        "black",
        "flake8",
        "matplotlib",
        "mypy",
    ],
}
EXTRAS_REQUIRE["all"] = sorted(
    {dependency for dependencies in EXTRAS_REQUIRE.values() for dependency in dependencies}
)

setup(
    name="waveflow",
    version="2.0.4",
    description="Waveflow: wireless propagation, waveform, and RIS-assisted network simulator",
    install_requires=BASE_REQUIRES,
    extras_require=EXTRAS_REQUIRE,
    packages=find_packages(
        include=[
            "app*",
            "cli*",
            "config*",
            "controller*",
            "core*",
            "risnet*",
            "waveflow*",
            "utils*",
        ],
        exclude=[
            "tests*",
            "examples*",
            "docs*",
        ],
    ),
    entry_points={
        "console_scripts": [
            "waveflow=waveflow.__main__:main",
            "risnet=risnet.__main__:main",
        ],
    },
)