Secure-Text-Classification/install.py at main · WiktorPieklik/Secure-Text-Classification · 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
from typing import List
from sys import platform, executable
from subprocess import check_call

common = [
    'numpy',
    'pandas',
    'jupyter',
    'matplotlib',
    'scikit-learn',
    'tensorflow-addons',
    'livelossplot',
]

windows = [
    # add any windows specific packages here
]

darwin = [
    'tensorflow-macos',
    'tensorflow-metal'
]

linux = [
    # add any linux specific packages here
]


def install(packages: List[str]) -> None:
    for package in packages:
        check_call([executable, '-m', 'pip', 'install', '--no-cache-dir', package])


if __name__ == '__main__':
    install(common)
    if platform == 'windows':
        install(windows)
    elif platform == 'darwin':
        install(darwin)
    elif platform == 'linux':
        install(linux)