-
Notifications
You must be signed in to change notification settings - Fork 52
/
setup.py
49 lines (41 loc) · 1.5 KB
/
setup.py
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
from setuptools import setup, find_packages
__version__ = '1.1.2'
URL = 'https://github.com/tczhangzhi/torcheeg'
install_requires = [
'tqdm>=4.64.0', 'numpy>=1.21.5', 'pandas>=1.3.5', 'xlrd>=2.0.1',
'scipy>=1.7.3, <= 1.10.1', 'scikit-learn>=1.0.2', 'lmdb>=1.3.0', 'einops>=0.4.1',
'mne>=1.0.3', 'xmltodict>=0.13.0', 'networkx>=2.6.3', 'PyWavelets>=1.3.0',
'spectrum>=0.8.1', 'torchmetrics>=0.10.0', 'mne_connectivity>=0.4.0',
'pytorch-lightning>=1.9.5'
]
test_requires = ['pytest>=7.1.2']
example_requires = []
pyg_requires = [
'torch-scatter',
'torch-sparse',
'torch-cluster',
'torch-spline-conv',
'torch_geometric>=2.0.3',
]
moabb_requires = ['moabb>=0.5.0']
readme = open('README.rst').read()
setup(
name='torcheeg',
version='1.1.2',
description=
'TorchEEG is a library built on PyTorch for EEG signal analysis. TorchEEG aims to provide a plug-and-play EEG analysis tool, so that researchers can quickly reproduce EEG analysis work and start new EEG analysis research without paying attention to technical details unrelated to the research focus.',
license='MIT',
author='TorchEEG Team',
author_email='tczhangzhi@gmail.com',
keywords=['PyTorch', 'EEG'],
url=URL,
packages=find_packages(),
long_description=readme,
python_requires='>=3.7',
extras_require={
'example': example_requires,
'test': test_requires,
'pyg': pyg_requires,
'moabb': moabb_requires
},
install_requires=install_requires)