pRNN/setup.py at main · LevensteinLab/pRNN · 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
from setuptools import setup, find_packages

# dependencies = [
#     "pandas==1.5.3",
#     "matplotlib==3.5.3",
#     "scipy==1.12.0",
#     "numpy==1.22.4",
#     "scikit-learn==1.4.0",
#     "pynapple==0.6.1",
#     "zipp==3.17.0",
#     "pyparsing==3.1.1",
#     "importlib_metadata==7.0.1",
#     "ruamel.yaml==0.18.6",
#     #"gym==0.21.0 --no-binary :all:", TODO: Need to fix this
#     "gymnasium==0.29.1",
#     "ratinabox==1.7.1"
# ]

dependencies = [
    "pandas",
    "matplotlib",
    "scipy",
    "numpy<2",
    "scikit-learn",
    "pynapple==0.6.1",
    "zipp",
    "pyparsing",
    "importlib_metadata",
    "ruamel.yaml",
    "gymnasium==0.29.1",
    "pytest==8.4.2",
]

setup(
    author="Daniel Levenstein",
    author_email="daniel.levenstein@mila.quebec",
    python_requires=">=3.9",
    name="prnn",
    version="v0.1",
    packages=find_packages(),
    install_requires=dependencies,
    description="Python Library for Predictive RNNs Modeling Hippocampal Representation and Replay",
    license="MIT License",
    # Add more metadata like author, description, etc. as needed
)