deepSTRF/setup.py at develop · urancon/deepSTRF · 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
from setuptools import setup, find_packages
from deepSTRF._version import __version__

setup(
    name='deepSTRF',
    version=__version__,
    description='A PyTorch-based library and benchmark for fitting sensory neural responses with deep neural network models',
    url="https://github.com/urancon/deepSTRF",
    author='Ulysse Rancon',
    author_email='[email protected]',
    license="GPL-3.0",
    packages=find_packages(include=['deepSTRF',
                                    'deepSTRF.datasets', 'deepSTRF.datasets.audio', 'deepSTRF.datasets.video',
                                    'deepSTRF.models', 'deepSTRF.models.audio', 'deepSTRF.models.video',
                                    'deepSTRF.metrics',
                                    'deepSTRF.utils',
                                    ]),
    install_requires=[
        'numpy==1.23.5',
        'scipy==1.15.2',
        'scikit-image',
        'torch==2.5.1',
        'torchaudio==2.5.1',
        'soundfile',
        'pytorch_lightning==2.5.1',
        'matplotlib==3.10.1',
        'Pillow==11.1.0',
        'torchvision',
        'DCLS==0.1.1',
        'einops==0.8.1',
        'h5py',
        'tables',
        'pandas',
        'wandb',
        'tqdm',
        'pykeops'
    ],
    extras_require={
        'allen': ['allensdk', 'xarray'],
        'nems': ['PyNEMS']
    }
)